xref: /llvm-project/mlir/test/Dialect/Tosa/variables.mlir (revision af972f01c01843a9ffe41ff496154267fa387a51) 
1// RUN: mlir-opt %s | mlir-opt | FileCheck %s
2// RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s
3
4
5// -----
6// CHECK-LABEL:   @test_variable_scalar(
7// CHECK-SAME:                        %[[ADD_VAL:.*]]: tensor<f32>) {
8func.func @test_variable_scalar(%arg0: tensor<f32>) -> () {
9  // CHECK:           tosa.variable @stored_var = dense<3.140000e+00> : tensor<f32>
10  tosa.variable @stored_var = dense<3.14> : tensor<f32>
11  // CHECK:           %[[STORED_VAL:.*]] = tosa.variable.read @stored_var : tensor<f32>
12  %0 = tosa.variable.read @stored_var : tensor<f32>
13  // CHECK:           %[[RESULT_ADD:.*]] = tosa.add %[[ADD_VAL]], %[[STORED_VAL]] : (tensor<f32>, tensor<f32>) -> tensor<f32>
14  %1 = "tosa.add"(%arg0, %0) : (tensor<f32>, tensor<f32>) -> tensor<f32>
15  // CHECK:           tosa.variable.write @stored_var, %[[RESULT_ADD]] : tensor<f32>
16  tosa.variable.write @stored_var, %1 : tensor<f32>
17  return
18}
19
20// -----
21// CHECK-LABEL:   @test_variable_tensor(
22// CHECK-SAME:                        %[[ADD_VAL:.*]]: tensor<2x4x8xi32>) {
23func.func @test_variable_tensor(%arg0: tensor<2x4x8xi32>) -> () {
24  // CHECK:           tosa.variable @stored_var = dense<-1> : tensor<2x4x8xi32>
25  tosa.variable @stored_var = dense<-1> : tensor<2x4x8xi32>
26  // CHECK:           %[[STORED_VAL:.*]] = tosa.variable.read @stored_var : tensor<2x4x8xi32>
27  %0 = tosa.variable.read @stored_var : tensor<2x4x8xi32>
28  // CHECK:           %[[RESULT_ADD:.*]] = tosa.add %[[ADD_VAL]], %[[STORED_VAL]] : (tensor<2x4x8xi32>, tensor<2x4x8xi32>) -> tensor<2x4x8xi32>
29  %1 = "tosa.add"(%arg0, %0) : (tensor<2x4x8xi32>, tensor<2x4x8xi32>) -> tensor<2x4x8xi32>
30  // CHECK:           tosa.variable.write @stored_var, %[[RESULT_ADD]] : tensor<2x4x8xi32>
31  tosa.variable.write @stored_var, %1 : tensor<2x4x8xi32>
32  return
33}
34