| // RUN: iree-run-mlir --Xcompiler,iree-input-type=stablehlo --Xcompiler,iree-hal-target-device=local --function=get_state --Xcompiler,iree-hal-local-target-device-backends=vmvx %s | FileCheck %s --check-prefix=CHECK-GET-STATE |
| // RUN: iree-run-mlir --Xcompiler,iree-input-type=stablehlo --Xcompiler,iree-hal-target-device=local --function=inc --Xcompiler,iree-hal-local-target-device-backends=vmvx %s | FileCheck %s --check-prefix=CHECK-INC |
| // RUN: iree-run-mlir --Xcompiler,iree-input-type=stablehlo --Xcompiler,iree-hal-target-device=local --function=get_state --Xcompiler,iree-hal-local-target-device-backends=llvm-cpu %s | FileCheck %s --check-prefix=CHECK-GET-STATE |
| // RUN: iree-run-mlir --Xcompiler,iree-input-type=stablehlo --Xcompiler,iree-hal-target-device=local --function=inc --Xcompiler,iree-hal-local-target-device-backends=llvm-cpu %s | FileCheck %s --check-prefix=CHECK-INC |
| |
| module { |
| ml_program.global private mutable @counter(dense<2.0> : tensor<f32>): tensor<f32> |
| |
| // CHECK-GET-STATE: EXEC @get_state |
| func.func @get_state() -> tensor<f32> { |
| %0 = ml_program.global_load @counter : tensor<f32> |
| return %0 : tensor<f32> |
| } |
| // CHECK-GET-STATE: f32=2 |
| |
| // CHECK-INC: EXEC @inc |
| func.func @inc() -> tensor<f32> { |
| %0 = ml_program.global_load @counter : tensor<f32> |
| %c1 = arith.constant dense<1.0> : tensor<f32> |
| %1 = stablehlo.add %0, %c1 : tensor<f32> |
| ml_program.global_store @counter = %1 : tensor<f32> |
| %2 = ml_program.global_load @counter : tensor<f32> |
| return %2 : tensor<f32> |
| } |
| // CHECK-INC: f32=3 |
| } |
