Skip to content

Commit b7e79c8

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

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)