diff options
-rw-r--r-- | mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt b/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt index 234a0d82babe..939d7481f1a2 100644 --- a/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt +++ b/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt @@ -27,3 +27,11 @@ add_mlir_dialect_library(MLIRSparseTensorPipelines MLIRVectorToLLVM MLIRVectorTransforms ) + +if(MLIR_ENABLE_CUDA_RUNNER) + # Enable gpu-to-cubin pass. + target_compile_definitions(obj.MLIRSparseTensorPipelines + PRIVATE + MLIR_GPU_TO_CUBIN_PASS_ENABLE=1 + ) +endif() |