diff --git a/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt b/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt --- a/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt +++ b/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt @@ -27,3 +27,11 @@ 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()