[llvm] [SPIRV] Add support for the SPIR-V extension SPV_KHR_bfloat16 (PR #155645)

via llvm-commits llvm-commits at lists.llvm.org
Mon Sep 15 05:21:35 PDT 2025


================
@@ -1222,6 +1222,12 @@ static void AddDotProductRequirements(const MachineInstr &MI,
   }
 }
 
+static bool isBFloat16Type(const SPIRVType *TypeDef) {
+  return TypeDef && TypeDef->getNumOperands() == 3 &&
+         TypeDef->getOpcode() == SPIRV::OpTypeFloat &&
+         TypeDef->getOperand(2).getImm() == SPIRV::FPEncoding::BFloat16KHR;
----------------
YixingZhang007 wrote:

Thanks for the suggestion! I have added the check for bitwidth here.

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


More information about the llvm-commits mailing list