Skip to content

Commit db0cdb1

Browse files
committed
[NVPTX] expand fexp2 and flog2 for v2f32
1 parent fc1e81f commit db0cdb1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

llvm/lib/Target/NVPTX/NVPTXISelLowering.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -988,14 +988,16 @@ NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
988988
setFP16OperationAction(ISD::FEXP2, MVT::v2f16, Legal, Expand);
989989
setBF16OperationAction(ISD::FEXP2, MVT::bf16, Legal, Promote);
990990
setBF16OperationAction(ISD::FEXP2, MVT::v2bf16, Legal, Expand);
991+
setOperationAction(ISD::FEXP2, MVT::v2f32, Expand);
991992

992993
// FLOG2 supports f32 only
993994
// f16/bf16 types aren't supported, but they are promoted/expanded to f32.
994995
if (UseApproxLog2F32) {
995996
setOperationAction(ISD::FLOG2, MVT::f32, Legal);
996997
setOperationPromotedToType(ISD::FLOG2, MVT::f16, MVT::f32);
997998
setOperationPromotedToType(ISD::FLOG2, MVT::bf16, MVT::f32);
998-
setOperationAction(ISD::FLOG2, {MVT::v2f16, MVT::v2bf16}, Expand);
999+
setOperationAction(ISD::FLOG2, {MVT::v2f16, MVT::v2bf16, MVT::v2f32},
1000+
Expand);
9991001
}
10001002

10011003
setOperationAction(ISD::ADDRSPACECAST, {MVT::i32, MVT::i64}, Custom);

0 commit comments

Comments
 (0)