docs: housekeeping + memref dialect (EnzymeAD#846)

* docs: housekeeping + memref dialect

* docs: autoadd dialects

Author: avik-pal

docs/make.jl

@@ -3,6 +3,11 @@ using Documenter, DocumenterVitepress
 
 DocMeta.setdocmeta!(Reactant, :DocTestSetup, :(using Reactant); recursive=true)
 
+# Helper functions
+function first_letter_uppercase(str)
+    return uppercase(str[1]) * str[2:end]
+end
+
 # Generate examples
 
 using Literate
@@ -32,16 +37,15 @@ pages = [
     "API Reference" => [
         "Reactant API" => "api/api.md",
         "Ops" => "api/ops.md",
-        "Dialects" => [
-            "ArithOps" => "api/arith.md",
-            "Affine" => "api/affine.md",
-            "Builtin" => "api/builtin.md",
-            "Chlo" => "api/chlo.md",
-            "Enzyme" => "api/enzyme.md",
-            "Func" => "api/func.md",
-            "StableHLO" => "api/stablehlo.md",
-            "VHLO" => "api/vhlo.md",
-        ],
+        "Dialects" => sort!(
+            [
+                first_letter_uppercase(first(splitext(basename(file)))) =>
+                    joinpath("api/dialects", file) for
+                file in readdir(joinpath(@__DIR__, "src/api/dialects")) if
+                splitext(file)[2] == ".md"
+            ];
+            by=first,
+        ),
         "MLIR API" => "api/mlirc.md",
         "XLA" => "api/xla.md",
         "Internal API" => "api/internal.md",
@@ -56,14 +60,13 @@ makedocs(;
         Reactant.MLIR,
         Reactant.MLIR.API,
         Reactant.MLIR.IR,
-        Reactant.MLIR.Dialects.chlo,
-        Reactant.MLIR.Dialects.vhlo,
-        Reactant.MLIR.Dialects.stablehlo,
-        Reactant.MLIR.Dialects.enzyme,
-        Reactant.MLIR.Dialects.arith,
-        Reactant.MLIR.Dialects.func,
-        Reactant.MLIR.Dialects.affine,
-        Reactant.MLIR.Dialects.builtin,
+        filter(
+            Base.Fix2(isa, Module),
+            [
+                getproperty(Reactant.MLIR.Dialects, x) for
+                x in names(Reactant.MLIR.Dialects; all=true) if x != :Dialects
+            ],
+        )...,
     ],
     authors="William Moses <[email protected]>, Valentin Churavy <[email protected]>",
     sitename="Reactant.jl",

docs/src/.vitepress/config.mts

@@ -136,21 +136,23 @@ export default defineConfig({
             text: "MLIR Dialects",
             collapsed: false,
             items: [
-              { text: "ArithOps", link: "/api/arith" },
-              { text: "Affine", link: "/api/affine" },
-              { text: "Builtin", link: "/api/builtin" },
-              { text: "Chlo", link: "/api/chlo" },
-              { text: "Enzyme", link: "/api/enzyme" },
-              { text: "Func", link: "/api/func" },
-              { text: "StableHLO", link: "/api/stablehlo" },
-              { text: "VHLO", link: "/api/vhlo" },
-              { text: "GPU", link: "/api/gpu" },
-              { text: "LLVM", link: "/api/llvm" },
-              { text: "NVVM", link: "/api/nvvm" },
-              { text: "TPU", link: "/api/tpu" },
-              { text: "Triton", link: "/api/triton" },
-              { text: "Shardy", link: "/api/shardy" },
-              { text: "MPI", link: "/api/mpi" },
+              { text: "ArithOps", link: "/api/dialects/arith" },
+              { text: "Affine", link: "/api/dialects/affine" },
+              { text: "Builtin", link: "/api/dialects/builtin" },
+              { text: "Chlo", link: "/api/dialects/chlo" },
+              { text: "Enzyme", link: "/api/dialects/enzyme" },
+              { text: "EnzymeXLA", link: "/api/dialects/enzymexla" },
+              { text: "Func", link: "/api/dialects/func" },
+              { text: "GPU", link: "/api/dialects/gpu" },
+              { text: "LLVM", link: "/api/dialects/llvm" },
+              { text: "MPI", link: "/api/dialects/mpi" },
+              { text: "MemRef", link: "/api/dialects/memref" },
+              { text: "NVVM", link: "/api/dialects/nvvm" },
+              { text: "Shardy", link: "/api/dialects/shardy" },
+              { text: "StableHLO", link: "/api/dialects/stablehlo" },
+              { text: "Triton", link: "/api/dialects/triton" },
+              { text: "TPU", link: "/api/dialects/tpu" },
+              { text: "VHLO", link: "/api/dialects/vhlo" },
             ],
           },
           {

docs/src/api/affine.md renamed to docs/src/api/dialects/affine.md

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