Refactor

raviqqe · Dec 4, 2023 · 66d4ad1 · 66d4ad1
1 parent d352179
commit 66d4ad1
Show file tree

Hide file tree

Showing 5 changed files with 64 additions and 49 deletions.
diff --git a/melior/src/pass/async.rs b/melior/src/pass/async.rs
@@ -1,10 +1,13 @@
 //! Async passes.
 
-melior_macro::async_passes!(
-    mlirCreateAsyncAsyncFuncToAsyncRuntime,
-    mlirCreateAsyncAsyncParallelFor,
-    mlirCreateAsyncAsyncRuntimePolicyBasedRefCounting,
-    mlirCreateAsyncAsyncRuntimeRefCounting,
-    mlirCreateAsyncAsyncRuntimeRefCountingOpt,
-    mlirCreateAsyncAsyncToAsyncRuntime,
+melior_macro::passes!(
+    "Async",
+    [
+        mlirCreateAsyncAsyncFuncToAsyncRuntime,
+        mlirCreateAsyncAsyncParallelFor,
+        mlirCreateAsyncAsyncRuntimePolicyBasedRefCounting,
+        mlirCreateAsyncAsyncRuntimeRefCounting,
+        mlirCreateAsyncAsyncRuntimeRefCountingOpt,
+        mlirCreateAsyncAsyncToAsyncRuntime,
+    ]
 );
diff --git a/melior/src/pass/gpu.rs b/melior/src/pass/gpu.rs
@@ -1,9 +1,12 @@
 //! GPU passes.
 
-melior_macro::gpu_passes!(
-    // spell-checker: disable-next-line
-    mlirCreateGPUGpuAsyncRegionPass,
-    mlirCreateGPUGpuKernelOutlining,
-    mlirCreateGPUGpuLaunchSinkIndexComputations,
-    mlirCreateGPUGpuMapParallelLoopsPass,
+melior_macro::passes!(
+    "GPU",
+    [
+        // spell-checker: disable-next-line
+        mlirCreateGPUGpuAsyncRegionPass,
+        mlirCreateGPUGpuKernelOutlining,
+        mlirCreateGPUGpuLaunchSinkIndexComputations,
+        mlirCreateGPUGpuMapParallelLoopsPass,
+    ]
 );
diff --git a/melior/src/pass/linalg.rs b/melior/src/pass/linalg.rs
@@ -1,15 +1,18 @@
 //! Linalg passes.
 
-melior_macro::linalg_passes!(
-    mlirCreateLinalgConvertElementwiseToLinalg,
-    mlirCreateLinalgLinalgBufferize,
-    mlirCreateLinalgLinalgDetensorize,
-    mlirCreateLinalgLinalgElementwiseOpFusion,
-    mlirCreateLinalgLinalgFoldUnitExtentDims,
-    mlirCreateLinalgLinalgGeneralization,
-    mlirCreateLinalgLinalgInlineScalarOperands,
-    mlirCreateLinalgLinalgLowerToAffineLoops,
-    mlirCreateLinalgLinalgLowerToLoops,
-    mlirCreateLinalgLinalgLowerToParallelLoops,
-    mlirCreateLinalgLinalgNamedOpConversion,
+melior_macro::passes!(
+    "Linalg",
+    [
+        mlirCreateLinalgConvertElementwiseToLinalg,
+        mlirCreateLinalgLinalgBufferize,
+        mlirCreateLinalgLinalgDetensorize,
+        mlirCreateLinalgLinalgElementwiseOpFusion,
+        mlirCreateLinalgLinalgFoldUnitExtentDims,
+        mlirCreateLinalgLinalgGeneralization,
+        mlirCreateLinalgLinalgInlineScalarOperands,
+        mlirCreateLinalgLinalgLowerToAffineLoops,
+        mlirCreateLinalgLinalgLowerToLoops,
+        mlirCreateLinalgLinalgLowerToParallelLoops,
+        mlirCreateLinalgLinalgNamedOpConversion,
+    ]
 );
diff --git a/melior/src/pass/sparse_tensor.rs b/melior/src/pass/sparse_tensor.rs
@@ -1,12 +1,15 @@
 //! Sparse tensor passes.
 
-melior_macro::sparse_tensor_passes!(
-    mlirCreateSparseTensorPostSparsificationRewrite,
-    mlirCreateSparseTensorPreSparsificationRewrite,
-    mlirCreateSparseTensorSparseBufferRewrite,
-    mlirCreateSparseTensorSparseTensorCodegen,
-    mlirCreateSparseTensorSparseTensorConversionPass,
-    mlirCreateSparseTensorSparseVectorization,
-    mlirCreateSparseTensorSparsificationPass,
-    mlirCreateSparseTensorStorageSpecifierToLLVM,
+melior_macro::passes!(
+    "SparseTensor",
+    [
+        mlirCreateSparseTensorPostSparsificationRewrite,
+        mlirCreateSparseTensorPreSparsificationRewrite,
+        mlirCreateSparseTensorSparseBufferRewrite,
+        mlirCreateSparseTensorSparseTensorCodegen,
+        mlirCreateSparseTensorSparseTensorConversionPass,
+        mlirCreateSparseTensorSparseVectorization,
+        mlirCreateSparseTensorSparsificationPass,
+        mlirCreateSparseTensorStorageSpecifierToLLVM,
+    ]
 );
diff --git a/melior/src/pass/transform.rs b/melior/src/pass/transform.rs
@@ -1,18 +1,21 @@
 //! Transform passes.
 
-melior_macro::transform_passes!(
-    mlirCreateTransformsCSE,
-    mlirCreateTransformsCanonicalizer,
-    mlirCreateTransformsControlFlowSink,
-    mlirCreateTransformsGenerateRuntimeVerification,
-    mlirCreateTransformsInliner,
-    mlirCreateTransformsLocationSnapshot,
-    mlirCreateTransformsLoopInvariantCodeMotion,
-    mlirCreateTransformsPrintOpStats,
-    mlirCreateTransformsSCCP,
-    mlirCreateTransformsStripDebugInfo,
-    mlirCreateTransformsSymbolDCE,
-    mlirCreateTransformsSymbolPrivatize,
-    mlirCreateTransformsTopologicalSort,
-    mlirCreateTransformsViewOpGraph,
+melior_macro::passes!(
+    "Transforms",
+    [
+        mlirCreateTransformsCSE,
+        mlirCreateTransformsCanonicalizer,
+        mlirCreateTransformsControlFlowSink,
+        mlirCreateTransformsGenerateRuntimeVerification,
+        mlirCreateTransformsInliner,
+        mlirCreateTransformsLocationSnapshot,
+        mlirCreateTransformsLoopInvariantCodeMotion,
+        mlirCreateTransformsPrintOpStats,
+        mlirCreateTransformsSCCP,
+        mlirCreateTransformsStripDebugInfo,
+        mlirCreateTransformsSymbolDCE,
+        mlirCreateTransformsSymbolPrivatize,
+        mlirCreateTransformsTopologicalSort,
+        mlirCreateTransformsViewOpGraph,
+    ]
 );