Fix issue to support fp64

intel · Apr 29, 2024 · 90e1152 · 90e1152
1 parent ffc18fa
commit 90e1152
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/third_party/intel/lib/TritonIntelGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandDPAS.cpp b/third_party/intel/lib/TritonIntelGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandDPAS.cpp
@@ -235,8 +235,11 @@ Type getSharedMemTy(Type argType) {
     return type::f32Ty(ctx);
   else if (argType.getIntOrFloatBitWidth() == 8)
     return type::i8Ty(ctx);
+  else if (argType.isF64())
+    return type::f64Ty(ctx);
   else
-    llvm::report_fatal_error("mma16816 data type not supported");
+    llvm::report_fatal_error(
+        "unsupported data type for the dot layout of DPAS");
 }
 
 template <unsigned opIdx>