diff --git a/linalg_ops/convolution/CMakeLists.txt b/linalg_ops/convolution/CMakeLists.txt index 0ac4900..ac06de4 100644 --- a/linalg_ops/convolution/CMakeLists.txt +++ b/linalg_ops/convolution/CMakeLists.txt @@ -75,6 +75,7 @@ foreach(_DTYPE_AND_LAYOUT IN LISTS _DTYPES_AND_LAYOUTS) "local-task" COMPILER_FLAGS "--iree-preprocessing-pass-pipeline=builtin.module\(func.func\(iree-linalg-ext-convert-conv2d-to-winograd{replace-all-convs=true}\)\)" + "--iree-llvmcpu-target-cpu=host" RUNNER_FLAGS LABELS "hostonly"