Small fixes (#1054)

rolfmorel · web-flow · commit d8755d3eb5a8 · 2025-05-28T13:25:48.000+01:00
diff --git a/include/TPP/Passes.td b/include/TPP/Passes.td
@@ -25,6 +25,7 @@ def LoadTppDialects : Pass<"load-tpp-dialects", "ModuleOp"> {
                            "check::CheckDialect",
                            "perf::PerfDialect",
                            "omp::OpenMPDialect",
+                           "amx::AMXDialect",
                            "async::AsyncDialect"];
 }
 
diff --git a/lib/TPP/LoadTppDialects.cpp b/lib/TPP/LoadTppDialects.cpp
@@ -15,6 +15,7 @@
 #include "TPP/Dialect/Xsmm/XsmmDialect.h"
 #include "mlir/Dialect/Async/IR/Async.h"
 #include "mlir/Dialect/OpenMP/OpenMPDialect.h"
+#include "mlir/Dialect/AMX/AMXDialect.h"
 
 
 namespace mlir {
diff --git a/python/mlir/dialects/perf.py b/python/mlir/dialects/perf.py
@@ -1,4 +1,4 @@
-from ._check_ops_gen import *
+from ._perf_ops_gen import *
 from .._mlir_libs._tppDialects.perf import *
 
 from .._mlir_libs import get_dialect_registry
diff --git a/python/mlir/dialects/xsmm.py b/python/mlir/dialects/xsmm.py
@@ -1,4 +1,4 @@
-from ._check_ops_gen import *
+from ._xsmm_ops_gen import *
 from .._mlir_libs._tppDialects.xsmm import *
 
 from .._mlir_libs import get_dialect_registry

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,7 @@ def LoadTppDialects : Pass<"load-tpp-dialects", "ModuleOp"> {`
`25`	`25`	`"check::CheckDialect",`
`26`	`26`	`"perf::PerfDialect",`
`27`	`27`	`"omp::OpenMPDialect",`
	`28`	`+ "amx::AMXDialect",`
`28`	`29`	`"async::AsyncDialect"];`
`29`	`30`	`}`
`30`	`31`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from ._check_ops_gen import *`
	`1`	`+from ._perf_ops_gen import *`
`2`	`2`	`from .._mlir_libs._tppDialects.perf import *`
`3`	`3`
`4`	`4`	`from .._mlir_libs import get_dialect_registry`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from ._check_ops_gen import *`
	`1`	`+from ._xsmm_ops_gen import *`
`2`	`2`	`from .._mlir_libs._tppDialects.xsmm import *`
`3`	`3`
`4`	`4`	`from .._mlir_libs import get_dialect_registry`