[Mlir-commits] [mlir] 195621a - [mlir][sparse][gpu] set cubin flag when building for cuda

[Aart Bik]

Author: Aart Bik
Date: 2023-05-16T10:50:22-07:00
New Revision: 195621aacb96a4e2f49f5d888e2fc7af20e6e797

URL: https://github.com/llvm/llvm-project/commit/195621aacb96a4e2f49f5d888e2fc7af20e6e797
DIFF: https://github.com/llvm/llvm-project/commit/195621aacb96a4e2f49f5d888e2fc7af20e6e797.diff

LOG: [mlir][sparse][gpu] set cubin flag when building for cuda

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D150692

Added: 
    

Modified: 
    mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt b/mlir/lib/Dialect/SparseTensor/Pipelines/CMakeLists.txt
index 234a0d82babef..939d7481f1a22 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()