@@ -992,14 +992,16 @@ NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
992992 setFP16OperationAction (ISD::FEXP2, MVT::v2f16, Legal, Expand);
993993 setBF16OperationAction (ISD::FEXP2, MVT::bf16 , Legal, Promote);
994994 setBF16OperationAction (ISD::FEXP2, MVT::v2bf16, Legal, Expand);
995+ setOperationAction (ISD::FEXP2, MVT::v2f32, Expand);
995996
996997 // FLOG2 supports f32 only
997998 // f16/bf16 types aren't supported, but they are promoted/expanded to f32.
998999 if (UseApproxLog2F32) {
9991000 setOperationAction (ISD::FLOG2, MVT::f32 , Legal);
10001001 setOperationPromotedToType (ISD::FLOG2, MVT::f16 , MVT::f32 );
10011002 setOperationPromotedToType (ISD::FLOG2, MVT::bf16 , MVT::f32 );
1002- setOperationAction (ISD::FLOG2, {MVT::v2f16, MVT::v2bf16}, Expand);
1003+ setOperationAction (ISD::FLOG2, {MVT::v2f16, MVT::v2bf16, MVT::v2f32},
1004+ Expand);
10031005 }
10041006
10051007 setOperationAction (ISD::ADDRSPACECAST, {MVT::i32 , MVT::i64 }, Custom);
0 commit comments