[Mlir-commits] [mlir] [MLIR] [AMX] Utilize x86_amx type for AMX dialect in MLIR. (PR #111197)

Ilya Enkovich llvmlistbot at llvm.org
Tue Nov 5 07:20:53 PST 2024


================
@@ -14,16 +14,20 @@ namespace mlir {
 class LLVMConversionTarget;
 class LLVMTypeConverter;
 class RewritePatternSet;
+class DialectRegistry;
 
 /// Collect a set of patterns to lower AMX ops to ops that map to LLVM
 /// intrinsics.
-void populateAMXLegalizeForLLVMExportPatterns(
-    const LLVMTypeConverter &converter, RewritePatternSet &patterns);
+void populateAMXLegalizeForLLVMExportPatterns(LLVMTypeConverter &converter,
----------------
ienkovich wrote:

This is because I now add `amx::TileType` conversion into the passed converter in `populateAMXLegalizeForLLVMExportPatterns`. I've found a similar trick used for ARM SME extension in `populateArmSMEToLLVMConversionPatterns` method.

https://github.com/llvm/llvm-project/pull/111197


More information about the Mlir-commits mailing list