Skip to content

Commit baaac0e

Browse files
committed
[NVPTX] only legalze fadd, fsub, fmul, fma for v2f32 on sm_100+
1 parent e5c402d commit baaac0e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llvm/lib/Target/NVPTX/NVPTXISelLowering.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -883,8 +883,8 @@ NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
883883
setBF16OperationAction(Op, MVT::bf16, Legal, Promote);
884884
if (getOperationAction(Op, MVT::bf16) == Promote)
885885
AddPromotedToType(Op, MVT::bf16, MVT::f32);
886-
if (STI.hasF32x2Instructions())
887-
setOperationAction(Op, MVT::v2f32, Legal);
886+
setOperationAction(Op, MVT::v2f32,
887+
STI.hasF32x2Instructions() ? Legal : Expand);
888888
}
889889

890890
// On SM80, we select add/mul/sub as fma to avoid promotion to float

0 commit comments

Comments
 (0)