gpu, llvm and nvvm dialect bindings (#315)

EnzymeAD · Jan 7, 2025 · ae305e3 · ae305e3
1 parent 26b9b70
commit ae305e3
Show file tree

Hide file tree

Showing 5 changed files with 6,755 additions and 0 deletions.
diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD
@@ -610,6 +610,48 @@ gentbl_cc_library(
     tblgen = "//:mlir-jl-tblgen",
 )
 
+gentbl_cc_library(
+    name = "LlvmJLIncGen",
+    tbl_outs = [(
+            ["--generator=jl-op-defs", "--disable-module-wrap=0"],
+            "Llvm.jl"
+        )
+    ],
+    td_file = "@llvm-project//mlir:include/mlir/Dialect/LLVMIR/LLVMOps.td",
+    deps = [
+        "@llvm-project//mlir:LLVMOpsTdFiles",
+    ],
+    tblgen = "//:mlir-jl-tblgen",
+)
+
+gentbl_cc_library(
+    name = "NvvmIncJLGen",
+    tbl_outs = [(
+            ["--generator=jl-op-defs", "--disable-module-wrap=0"],
+            "Nvvm.jl"
+        )
+    ],
+    td_file = "@llvm-project//mlir:include/mlir/Dialect/LLVMIR/NVVMOps.td",
+    deps = [
+        "@llvm-project//mlir:NVVMOpsTdFiles",
+    ],
+    tblgen = "//:mlir-jl-tblgen",
+)
+
+gentbl_cc_library(
+    name = "GpuIncJLGen",
+    tbl_outs = [(
+            ["--generator=jl-op-defs", "--disable-module-wrap=0"],
+            "Gpu.jl"
+        )
+    ],
+    td_file = "@llvm-project//mlir:include/mlir/Dialect/GPU/IR/GPUOps.td",
+    deps = [
+        "@llvm-project//mlir:GPUOpsTdFiles",
+    ],
+    tblgen = "//:mlir-jl-tblgen",
+)
+
 gentbl_cc_library(
     name = "EnzymeJLIncGen",
     tbl_outs = [(

diff --git a/deps/ReactantExtra/make-bindings.jl b/deps/ReactantExtra/make-bindings.jl
@@ -23,6 +23,10 @@ for file in [
     "StableHLO.jl",
     "CHLO.jl",
     "VHLO.jl",
+    "Llvm.jl",
+    "Nvvm.jl",
+    "Gpu.jl",
+    "Affine.jl",
 ]
     build_file(joinpath(src_dir, "mlir", "Dialects", file))
 end