xref: /llvm-project/mlir/test/Integration/Dialect/Vector/CPU/X86Vector/dot.mlir (revision 435114f9fe2139bec770e5a95799f4eab20639e7)
1// RUN: mlir-opt %s -convert-vector-to-scf -convert-scf-to-cf -convert-vector-to-llvm="enable-x86vector" -convert-func-to-llvm -reconcile-unrealized-casts | \
2// RUN: mlir-translate --mlir-to-llvmir | \
3// RUN: %lli --entry-function=entry --mattr="avx" --dlopen=%mlir_c_runner_utils | \
4// RUN: FileCheck %s
5
6func.func @entry() -> i32 {
7  %i0 = arith.constant 0 : i32
8  %i4 = arith.constant 4 : i32
9
10  %a = arith.constant dense<[1.0,  2.0,  3.0,  4.0,  5.0,  6.0,  7.0,  8.0]> : vector<8xf32>
11  %b = arith.constant dense<[9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0]> : vector<8xf32>
12  %r = x86vector.avx.intr.dot %a, %b : vector<8xf32>
13
14  %1 = vector.extractelement %r[%i0 : i32]: vector<8xf32>
15  %2 = vector.extractelement %r[%i4 : i32]: vector<8xf32>
16  %d = arith.addf %1, %2 : f32
17
18  // CHECK: ( 110, 110, 110, 110, 382, 382, 382, 382 )
19  // CHECK: 492
20  vector.print %r : vector<8xf32>
21  vector.print %d : f32
22
23  return %i0 : i32
24}
25