Skip to content

Commit cc7abe0

Browse files
committed
[NVPTX] expand fexp2 and flog2 for v2f32
1 parent 1b24165 commit cc7abe0

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
@@ -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

Comments
 (0)