
// RUN: iree-compile %s \
// RUN:   --iree-hal-target-device=local \
// RUN:   --iree-hal-local-target-device-backends=vmvx \
// RUN:   --iree-opt-splat-parameters=%t.irpa | \
// RUN: iree-run-module \
// RUN:   --device=local-sync \
// RUN:   --module=- \
// RUN:   --function=main \
// RUN:   --parameters=scope=%t.irpa \
// RUN:   --input=1x2xi32=1 | \
// RUN: FileCheck %s

// CHECK-LABEL: EXEC @main
// CHECK: 1x2xi32=[0 0]

util.global private @array_global_0 = #flow.parameter.named<"scope"::"global_0"> : tensor<1x2xi32>
util.global private @dense_global_1 = #flow.parameter.named<"scope"::"global_1"> : tensor<2x2xi32>
util.global private @dense_global_2 = #flow.parameter.named<"scope"::"global_2"> : tensor<1x2xi32>
util.global private @dense_global_3 = #flow.parameter.named<"scope"::"global_3"> : tensor<2x2xi32>
func.func @main(%arg0: tensor<1x2xi32>) -> tensor<1x2xi32> {
  %cst = arith.constant 0 : i32
  %3 = util.global.load @array_global_0 : tensor<1x2xi32>
  %4 = util.global.load @dense_global_1 : tensor<2x2xi32>
  %5 = util.global.load @dense_global_2 : tensor<1x2xi32>
  %6 = util.global.load @dense_global_3 : tensor<2x2xi32>
  %empty = tensor.empty() : tensor<1x2xi32>
  %fill = linalg.fill ins(%cst : i32) outs(%empty : tensor<1x2xi32>) -> tensor<1x2xi32>
  %8 = linalg.matmul ins(%arg0, %6 : tensor<1x2xi32>, tensor<2x2xi32>) outs(%fill : tensor<1x2xi32>) -> tensor<1x2xi32>
  %10 = linalg.add ins(%8, %5 : tensor<1x2xi32>, tensor<1x2xi32>) outs(%empty : tensor<1x2xi32>) -> tensor<1x2xi32>
  %12 = linalg.matmul ins(%10, %4 : tensor<1x2xi32>, tensor<2x2xi32>) outs(%fill : tensor<1x2xi32>) -> tensor<1x2xi32>
  %14 = linalg.add ins(%12, %3 : tensor<1x2xi32>, tensor<1x2xi32>) outs(%empty : tensor<1x2xi32>) -> tensor<1x2xi32>
  return %14 : tensor<1x2xi32>
}