[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