Add support for i32 to SharedToDotOperandMMAv2

acollins3 · acollins3 · commit 4ea1cf55132d · 2024-09-05T11:53:20.000Z
diff --git a/third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandMMAv2.cpp b/third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandMMAv2.cpp
@@ -498,6 +498,8 @@ Type getSharedMemTy(Type argType) {
     return type::i8Ty(ctx);
   else if (argType.getIntOrFloatBitWidth() == 16)
     return type::i16Ty(ctx);
+  else if (argType.getIntOrFloatBitWidth() == 32)
+    return type::i32Ty(ctx);
   else
     llvm::report_fatal_error("mma16816 data type not supported");
 }

Original file line number	Diff line number	Diff line change
`@@ -498,6 +498,8 @@ Type getSharedMemTy(Type argType) {`
`498`	`498`	`return type::i8Ty(ctx);`
`499`	`499`	`else if (argType.getIntOrFloatBitWidth() == 16)`
`500`	`500`	`return type::i16Ty(ctx);`
	`501`	`+ else if (argType.getIntOrFloatBitWidth() == 32)`
	`502`	`+ return type::i32Ty(ctx);`
`501`	`503`	`else`
`502`	`504`	`llvm::report_fatal_error("mma16816 data type not supported");`
`503`	`505`	`}`