[llvm] [NVPTX] Support BFloat Store Parameter (PR #137074)

Alex MacLean via llvm-commits llvm-commits at lists.llvm.org
Thu Apr 24 08:14:44 PDT 2025


================
@@ -1852,7 +1852,8 @@ bool NVPTXDAGToDAGISel::tryStoreParam(SDNode *N) {
     case 1: {
       MVT::SimpleValueType MemTy = Mem->getMemoryVT().getSimpleVT().SimpleTy;
       SDValue Imm = Ops[0];
-      if (MemTy != MVT::f16 && MemTy != MVT::v2f16 &&
+      if (MemTy != MVT::f16 && MemTy != MVT::v2f16 && MemTy != MVT::bf16 &&
----------------
AlexMaclean wrote:

I think we can probably just remove the vector types here. A ConstantFPSDNode should never have one of these types. 

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


More information about the llvm-commits mailing list