feat: SparseTensor dialect (#1218)

* feat: SparseTensor dialect

* docs: add to docs

Author: avik-pal

deps/ReactantExtra/BUILD

@@ -921,6 +921,20 @@ gentbl_cc_library(
     includes = ["external/shardy"],
 )
 
+gentbl_cc_library(
+    name = "SparseTensorJLIncGen",
+    tbl_outs = [(
+            ["--generator=jl-op-defs", "--disable-module-wrap=0"],
+            "SparseTensor.jl"
+        )
+    ],
+    td_file = "@llvm-project//mlir:include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td",
+    deps = [
+        "@llvm-project//mlir:SparseTensorTdFiles",
+    ],
+    tblgen = "//:mlir-jl-tblgen",
+)
+
 genrule(
     name = "libMLIR_h.jl",
     tags = [

deps/ReactantExtra/make-bindings.jl

@@ -40,6 +40,7 @@ for file in [
     "Shardy.jl",
     "MPI.jl",
     "MemRef.jl",
+    "SparseTensor.jl",
 ]
     build_file(joinpath(src_dir, "mlir", "Dialects", file))
 end

docs/src/.vitepress/config.mts

@@ -111,6 +111,7 @@ export default defineConfig({
               { text: "MemRef", link: "/api/dialects/memref" },
               { text: "NVVM", link: "/api/dialects/nvvm" },
               { text: "Shardy", link: "/api/dialects/shardy" },
+              { text: "SparseTensor", link: "/api/dialects/sparsetensor" },
               { text: "StableHLO", link: "/api/dialects/stablehlo" },
               { text: "Triton", link: "/api/dialects/triton" },
               { text: "TPU", link: "/api/dialects/tpu" },
@@ -183,6 +184,7 @@ export default defineConfig({
               { text: "MemRef", link: "/api/dialects/memref" },
               { text: "NVVM", link: "/api/dialects/nvvm" },
               { text: "Shardy", link: "/api/dialects/shardy" },
+              { text: "SparseTensor", link: "/api/dialects/sparsetensor" },
               { text: "StableHLO", link: "/api/dialects/stablehlo" },
               { text: "Triton", link: "/api/dialects/triton" },
               { text: "TPU", link: "/api/dialects/tpu" },

docs/src/api/dialects/sparsetensor.md

@@ -0,0 +1,9 @@
+```@meta
+CollapsedDocStrings = true
+```
+
+# SparseTensor Dialect
+
+```@autodocs
+Modules = [Reactant.MLIR.Dialects.tt]
+```