Skip to content

Commit dfa0918

Browse files
committed
[NVPTX] expand fexp2 and flog2 for v2f32
1 parent 0e3ee1b commit dfa0918

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
@@ -986,14 +986,16 @@ NVPTXTargetLowering::NVPTXTargetLowering(const NVPTXTargetMachine &TM,
986986
setFP16OperationAction(ISD::FEXP2, MVT::v2f16, Legal, Expand);
987987
setBF16OperationAction(ISD::FEXP2, MVT::bf16, Legal, Promote);
988988
setBF16OperationAction(ISD::FEXP2, MVT::v2bf16, Legal, Expand);
989+
setOperationAction(ISD::FEXP2, MVT::v2f32, Expand);
989990

990991
// FLOG2 supports f32 only
991992
// f16/bf16 types aren't supported, but they are promoted/expanded to f32.
992993
if (UseApproxLog2F32) {
993994
setOperationAction(ISD::FLOG2, MVT::f32, Legal);
994995
setOperationPromotedToType(ISD::FLOG2, MVT::f16, MVT::f32);
995996
setOperationPromotedToType(ISD::FLOG2, MVT::bf16, MVT::f32);
996-
setOperationAction(ISD::FLOG2, {MVT::v2f16, MVT::v2bf16}, Expand);
997+
setOperationAction(ISD::FLOG2, {MVT::v2f16, MVT::v2bf16, MVT::v2f32},
998+
Expand);
997999
}
9981000

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

0 commit comments

Comments
 (0)