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

Steffi Stumpos via llvm-commits llvm-commits at lists.llvm.org
Thu Apr 24 15:02:10 PDT 2025


================
@@ -1868,7 +1868,7 @@ 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::bf16 &&
----------------
stumpOS wrote:

I removed it in response to Alex's comment under the belief that it was impossible to create a node of type ConstantSDNode or ConstantFPSDNode of vector type. If that assumption is incorrect I'm happy to add it back in. However, a test that fails without it and succeeds with it would be required. Can you provide the corner case so I can add a test?

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


More information about the llvm-commits mailing list