// RUN: mlir-opt %s | mlir-opt | FileCheck %s // RUN: mlir-opt %s --mlir-print-op-generic | mlir-opt | FileCheck %s // ----- // CHECK-LABEL: @test_variable_scalar( // CHECK-SAME: %[[ADD_VAL:.*]]: tensor) { func.func @test_variable_scalar(%arg0: tensor) -> () { // CHECK: tosa.variable @stored_var = dense<3.140000e+00> : tensor tosa.variable @stored_var = dense<3.14> : tensor // CHECK: %[[STORED_VAL:.*]] = tosa.variable.read @stored_var : tensor %0 = tosa.variable.read @stored_var : tensor // CHECK: %[[RESULT_ADD:.*]] = tosa.add %[[ADD_VAL]], %[[STORED_VAL]] : (tensor, tensor) -> tensor %1 = "tosa.add"(%arg0, %0) : (tensor, tensor) -> tensor // CHECK: tosa.variable.write @stored_var, %[[RESULT_ADD]] : tensor tosa.variable.write @stored_var, %1 : tensor return } // ----- // CHECK-LABEL: @test_variable_tensor( // CHECK-SAME: %[[ADD_VAL:.*]]: tensor<2x4x8xi32>) { func.func @test_variable_tensor(%arg0: tensor<2x4x8xi32>) -> () { // CHECK: tosa.variable @stored_var = dense<-1> : tensor<2x4x8xi32> tosa.variable @stored_var = dense<-1> : tensor<2x4x8xi32> // CHECK: %[[STORED_VAL:.*]] = tosa.variable.read @stored_var : tensor<2x4x8xi32> %0 = tosa.variable.read @stored_var : tensor<2x4x8xi32> // CHECK: %[[RESULT_ADD:.*]] = tosa.add %[[ADD_VAL]], %[[STORED_VAL]] : (tensor<2x4x8xi32>, tensor<2x4x8xi32>) -> tensor<2x4x8xi32> %1 = "tosa.add"(%arg0, %0) : (tensor<2x4x8xi32>, tensor<2x4x8xi32>) -> tensor<2x4x8xi32> // CHECK: tosa.variable.write @stored_var, %[[RESULT_ADD]] : tensor<2x4x8xi32> tosa.variable.write @stored_var, %1 : tensor<2x4x8xi32> return }