xref: /llvm-project/mlir/test/Dialect/Arith/emulate-wide-int-very-wide.mlir (revision 9816edc9f3ce198d41e364dd3467caa839a0c220)
1// Check that emulation of wery wide types (>64 bits) works as expected.
2
3// RUN: mlir-opt --arith-emulate-wide-int="widest-int-supported=512" %s | FileCheck %s
4
5// CHECK-LABEL: func.func @muli_scalar
6// CHECK-SAME:    ([[ARG0:%.+]]: vector<2xi512>, [[ARG1:%.+]]: vector<2xi512>) -> vector<2xi512>
7// CHECK-NEXT:    [[LOW0:%.+]]   = vector.extract [[ARG0]][0] : i512 from vector<2xi512>
8// CHECK-NEXT:    [[HIGH0:%.+]]  = vector.extract [[ARG0]][1] : i512 from vector<2xi512>
9// CHECK-NEXT:    [[LOW1:%.+]]   = vector.extract [[ARG1]][0] : i512 from vector<2xi512>
10// CHECK-NEXT:    [[HIGH1:%.+]]  = vector.extract [[ARG1]][1] : i512 from vector<2xi512>
11//
12// CHECK-DAG:     arith.mului_extended
13// CHECK-DAG:     arith.muli
14// CHECK-DAG:     arith.muli
15// CHECK-NEXT:    arith.addi
16// CHECK-NEXT:    arith.addi
17//
18// CHECK:         return {{%.+}} : vector<2xi512>
19func.func @muli_scalar(%a : i1024, %b : i1024) -> i1024 {
20    %m = arith.muli %a, %b : i1024
21    return %m : i1024
22}
23