diff --git a/third_party/intel/lib/TritonIntelGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandDPAS.cpp b/third_party/intel/lib/TritonIntelGPUToLLVM/ConvertLayoutOpToLLVM/SharedToDotOperandDPAS.cpp index f859551ef0..e0b1e319b7 100644 --- 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