Introduce unstructured-to-memref pass

include/triton-shared/Conversion/CMakeLists.txt

@@ -2,6 +2,7 @@ add_subdirectory(TritonToLinalg)
 add_subdirectory(TritonToLinalgExperimental)
 add_subdirectory(TritonToStructured)
 add_subdirectory(TritonArithToLinalg)
+add_subdirectory(TritonPtrToMemref)
 add_subdirectory(TritonToUnstructured)
 add_subdirectory(StructuredToMemref)
-add_subdirectory(TritonPtrToMemref)
+add_subdirectory(UnstructuredToMemref)

include/triton-shared/Conversion/UnstructuredToMemref/CMakeLists.txt

@@ -0,0 +1,10 @@
+#===------------------------------------------------------------------------===#
+#
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+#
+#===------------------------------------------------------------------------===#
+
+set(LLVM_TARGET_DEFINITIONS Passes.td)
+mlir_tablegen(Passes.h.inc -gen-pass-decls --name UnstructuredToMemref)
+add_public_tablegen_target(UnstructuredToMemrefConversionPassIncGen)

include/triton-shared/Conversion/UnstructuredToMemref/Passes.h

@@ -0,0 +1,22 @@
+//===----------------------------------------------------------------------===//
+//
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef UNSTRUCTURED_TO_MEMREF_CONVERSION_PASSES_H
+#define UNSTRUCTURED_TO_MEMREF_CONVERSION_PASSES_H
+
+#include "triton-shared/Conversion/UnstructuredToMemref/UnstructuredToMemref.h"
+
+namespace mlir {
+namespace triton {
+
+#define GEN_PASS_REGISTRATION
+#include "triton-shared/Conversion/UnstructuredToMemref/Passes.h.inc"
+
+} // namespace triton
+} // namespace mlir
+
+#endif

include/triton-shared/Conversion/UnstructuredToMemref/Passes.td

@@ -0,0 +1,18 @@
+//===----------------------------------------------------------------------===//
+//
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef UNSTRUCTURED_TO_MEMREF_CONVERSION_PASSES
+#define UNSTRUCTURED_TO_MEMREF_CONVERSION_PASSES
+
+include "mlir/Pass/PassBase.td"
+
+def UnstructuredToMemref : Pass<"unstructured-to-memref", "mlir::ModuleOp"> {
+  let summary = "Convert unstructured triton ptr (gather / scatter) to memref";
+  let constructor = "triton::createUnstructuredToMemrefPass()";
+}
+
+#endif

include/triton-shared/Conversion/UnstructuredToMemref/UnstructuredToMemref.h

@@ -0,0 +1,21 @@
+//===----------------------------------------------------------------------===//
+//
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef TRITON_CONVERSION_UNSTRUCTUREDTOMEMREF_UNSTRUCTUREDTOMEMREF_H
+#define TRITON_CONVERSION_UNSTRUCTUREDTOMEMREF_UNSTRUCTUREDTOMEMREF_H
+
+#include "mlir/Pass/Pass.h"
+
+namespace mlir {
+namespace triton {
+
+std::unique_ptr<OperationPass<ModuleOp>> createUnstructuredToMemrefPass();
+
+} // namespace triton
+} // namespace mlir
+
+#endif // TRITON_CONVERSION_UNSTRUCTUREDTOMEMREF_UNSTRUCTUREDTOMEMREF_H

lib/Conversion/CMakeLists.txt

@@ -5,3 +5,4 @@ add_subdirectory(TritonToUnstructured)
 add_subdirectory(TritonArithToLinalg)
 add_subdirectory(StructuredToMemref)
 add_subdirectory(TritonPtrToMemref)
+add_subdirectory(UnstructuredToMemref)

lib/Conversion/UnstructuredToMemref/CMakeLists.txt

@@ -0,0 +1,28 @@
+#===------------------------------------------------------------------------===#
+#
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT license.
+#
+#===------------------------------------------------------------------------===#
+
+add_triton_library(UnstructuredToMemref
+  UnstructuredToMemrefPass.cpp
+
+  DEPENDS
+  UnstructuredToMemrefConversionPassIncGen
+
+  LINK_LIBS PUBLIC
+  TritonTilingExtIR
+  MLIRArithDialect
+  MLIRDialectUtils
+  MLIRIR
+  MLIRMathDialect
+  MLIRPass
+  MLIRTensorDialect
+  MLIRTransforms
+  MLIRSupport
+  TritonAnalysis
+  TritonIR
+  TritonTransforms
+  TritonSharedAnalysis
+)