Skip to content

Commit 9e26258

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

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
@@ -847,8 +847,8 @@ NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
847847
setBF16OperationAction(Op, MVT::bf16, Legal, Promote);
848848
if (getOperationAction(Op, MVT::bf16) == Promote)
849849
AddPromotedToType(Op, MVT::bf16, MVT::f32);
850-
if (STI.hasF32x2Instructions())
851-
setOperationAction(Op, MVT::v2f32, Legal);
850+
setOperationAction(Op, MVT::v2f32,
851+
STI.hasF32x2Instructions() ? Legal : Expand);
852852
}
853853

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

0 commit comments

Comments
 (0)