1 | //===- AttrToLLVMConverter.cpp - Arith attributes conversion to LLVM ------===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | |
9 | #include "mlir/Conversion/ArithCommon/AttrToLLVMConverter.h" |
10 | |
11 | using namespace mlir; |
12 | |
13 | LLVM::FastmathFlags |
14 | mlir::arith::convertArithFastMathFlagsToLLVM(arith::FastMathFlags arithFMF) { |
15 | LLVM::FastmathFlags llvmFMF{}; |
16 | const std::pair<arith::FastMathFlags, LLVM::FastmathFlags> flags[] = { |
17 | {arith::FastMathFlags::nnan, LLVM::FastmathFlags::nnan}, |
18 | {arith::FastMathFlags::ninf, LLVM::FastmathFlags::ninf}, |
19 | {arith::FastMathFlags::nsz, LLVM::FastmathFlags::nsz}, |
20 | {arith::FastMathFlags::arcp, LLVM::FastmathFlags::arcp}, |
21 | {arith::FastMathFlags::contract, LLVM::FastmathFlags::contract}, |
22 | {arith::FastMathFlags::afn, LLVM::FastmathFlags::afn}, |
23 | {arith::FastMathFlags::reassoc, LLVM::FastmathFlags::reassoc}}; |
24 | for (auto [arithFlag, llvmFlag] : flags) { |
25 | if (bitEnumContainsAny(arithFMF, arithFlag)) |
26 | llvmFMF = llvmFMF | llvmFlag; |
27 | } |
28 | return llvmFMF; |
29 | } |
30 | |
31 | LLVM::FastmathFlagsAttr |
32 | mlir::arith::convertArithFastMathAttrToLLVM(arith::FastMathFlagsAttr fmfAttr) { |
33 | arith::FastMathFlags arithFMF = fmfAttr.getValue(); |
34 | return LLVM::FastmathFlagsAttr::get( |
35 | fmfAttr.getContext(), convertArithFastMathFlagsToLLVM(arithFMF)); |
36 | } |
37 | |
38 | LLVM::IntegerOverflowFlags mlir::arith::convertArithOverflowFlagsToLLVM( |
39 | arith::IntegerOverflowFlags arithFlags) { |
40 | LLVM::IntegerOverflowFlags llvmFlags{}; |
41 | const std::pair<arith::IntegerOverflowFlags, LLVM::IntegerOverflowFlags> |
42 | flags[] = { |
43 | {arith::IntegerOverflowFlags::nsw, LLVM::IntegerOverflowFlags::nsw}, |
44 | {arith::IntegerOverflowFlags::nuw, LLVM::IntegerOverflowFlags::nuw}}; |
45 | for (auto [arithFlag, llvmFlag] : flags) { |
46 | if (bitEnumContainsAny(arithFlags, arithFlag)) |
47 | llvmFlags = llvmFlags | llvmFlag; |
48 | } |
49 | return llvmFlags; |
50 | } |
51 | |
52 | LLVM::RoundingMode |
53 | mlir::arith::convertArithRoundingModeToLLVM(arith::RoundingMode roundingMode) { |
54 | switch (roundingMode) { |
55 | case arith::RoundingMode::downward: |
56 | return LLVM::RoundingMode::TowardNegative; |
57 | case arith::RoundingMode::to_nearest_away: |
58 | return LLVM::RoundingMode::NearestTiesToAway; |
59 | case arith::RoundingMode::to_nearest_even: |
60 | return LLVM::RoundingMode::NearestTiesToEven; |
61 | case arith::RoundingMode::toward_zero: |
62 | return LLVM::RoundingMode::TowardZero; |
63 | case arith::RoundingMode::upward: |
64 | return LLVM::RoundingMode::TowardPositive; |
65 | } |
66 | llvm_unreachable("Unhandled rounding mode" ); |
67 | } |
68 | |
69 | LLVM::RoundingModeAttr mlir::arith::convertArithRoundingModeAttrToLLVM( |
70 | arith::RoundingModeAttr roundingModeAttr) { |
71 | assert(roundingModeAttr && "Expecting valid attribute" ); |
72 | return LLVM::RoundingModeAttr::get( |
73 | roundingModeAttr.getContext(), |
74 | convertArithRoundingModeToLLVM(roundingModeAttr.getValue())); |
75 | } |
76 | |
77 | LLVM::FPExceptionBehaviorAttr |
78 | mlir::arith::getLLVMDefaultFPExceptionBehavior(MLIRContext &context) { |
79 | return LLVM::FPExceptionBehaviorAttr::get(&context, |
80 | LLVM::FPExceptionBehavior::Ignore); |
81 | } |
82 | |