| ### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ### | |
| iree_add_all_subdirs() | |
| iree_microbenchmark_suite( | |
| NAME | |
| "microbenchmark_llvm-cpu" | |
| SRCS | |
| "dynamic_shape_vectorization.mlir" | |
| "linalg_mmt4d.mlir" | |
| "linalg_transpose.mlir" | |
| "stablehlo_conv.mlir" | |
| "stablehlo_dot.mlir" | |
| "stablehlo_dot_general.mlir" | |
| "stablehlo_fft_abs.mlir" | |
| FLAGS | |
| "--iree-hal-target-backends=llvm-cpu" | |
| "--iree-input-type=stablehlo" | |
| "--iree-llvmcpu-target-cpu-features=host" | |
| ) | |
| # Only target cuda benchmarks if the backend is enabled. | |
| if(IREE_TARGET_BACKEND_CUDA) | |
| iree_microbenchmark_suite( | |
| NAME | |
| "microbenchmark_cuda" | |
| SRCS | |
| "linalg_transpose.mlir" | |
| FLAGS | |
| "--iree-hal-target-backends=cuda" | |
| "--iree-input-type=stablehlo" | |
| ) | |
| endif() |