Copy LinalgToXeGPU pass from tpp-mlir (#165)

* Copy LinalgToXeGPU pass from tpp-mlir

* fix clang-format

* fix clang-tidy

* fix build warnings

Signed-off-by: Dmitry Chigarev <dmitry.chigarev@intel.com>

* remove unused utils

Signed-off-by: Dmitry Chigarev <dmitry.chigarev@intel.com>

* licenses

Signed-off-by: Dmitry Chigarev <dmitry.chigarev@intel.com>

---------

Signed-off-by: Dmitry Chigarev <dmitry.chigarev@intel.com>

Author: dchigarev

include/gc/Transforms/Passes.h

@@ -36,6 +36,10 @@ namespace MemRef {
 class MemRefDialect;
 }
 
+namespace xegpu {
+class XeGPUDialect;
+}
+
 class PassManager;
 
 namespace gc {

include/gc/Transforms/Passes.td

@@ -46,4 +46,29 @@ def GCCPUPipeline: Pass<"gc-cpu-pipeline"> {
       "vector::VectorDialect"];
 }
 
+def LinalgToXeGPU : Pass<"linalg-to-xegpu", "func::FuncOp"> {
+  let summary = "Convert linalg dialect to XeGPU dialect.";
+  let description = [{
+    Lower linalg ops to XeGPU dialect.
+  }];
+  let dependentDialects = ["linalg::LinalgDialect",
+                           "gpu::GPUDialect",
+                           "xegpu::XeGPUDialect",
+                           "scf::SCFDialect",
+                           "memref::MemRefDialect",
+                           "arith::ArithDialect",
+                           "math::MathDialect",
+                           "vector::VectorDialect"];
+  let options = [
+    Option<"kTile", "k-tile", "int64_t",
+           /*default=*/"32",
+           "GEMM tile size for reduction dimension.">,
+    Option<"stages", "stages", "int64_t",
+           /*default=*/"1",
+           "Number of cooperative prefetch stages.">,
+    ListOption<"dpasTile", "dpas-tile", "int64_t",
+               "DPAS register block sizes MxNxK">,
+  ];
+}
+
 #endif // GC_DIALECT_GC_PASSES

include/gc/Transforms/Utils/MatcherUtils.h

@@ -0,0 +1,34 @@
+//===- MatcherUtils.h - Matcher utils ---------------------------*- C++ -*-===//
+//
+// This file is licensed under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef GC_MATCHERUTILS_H
+#define GC_MATCHERUTILS_H
+
+namespace mlir {
+class Value;
+namespace linalg {
+class LinalgOp;
+class GenericOp;
+} // namespace linalg
+namespace structured_match {
+namespace utils {
+
+// Returns true if the linalg operation is a 2d eltwsie floating point addition.
+bool isTwoDAddOp(linalg::LinalgOp linalgOp,
+                 SmallVectorImpl<Value> *capturedOperands = nullptr);
+
+// Returns true if the linalg.generic is a 2d eltwise floating point relu
+// operation.
+bool isTwoDReluOp(linalg::LinalgOp linalgOp,
+                  SmallVectorImpl<Value> *capturedOperands = nullptr);
+
+} // namespace utils
+} // namespace structured_match
+} // namespace mlir
+
+#endif // GC_MATCHERUTILS_H