// RUN: iree-run-mlir -iree-hal-target-backends=interpreter-bytecode %s | IreeFileCheck %s
// RUN: [[ $IREE_VULKAN_DISABLE == 1 ]] || (iree-run-mlir -iree-hal-target-backends=vulkan-spirv %s | IreeFileCheck %s)

module {
  flow.variable @counter mutable dense<2.0> : tensor<f32>

  // CHECK: EXEC @get_state
  func @get_state() -> tensor<f32> {
    %0 = flow.variable.load @counter : tensor<f32>
    return %0 : tensor<f32>
  }
  // CHECK: f32=2

  // CHECK: EXEC @inc
  func @inc() -> tensor<f32> {
    %0 = flow.variable.load @counter : tensor<f32>
    %c1 = constant dense<1.0> : tensor<f32>
    %1 = xla_hlo.add %0, %c1 : tensor<f32>
    flow.variable.store %1, @counter : tensor<f32>
    %2 = flow.variable.load @counter : tensor<f32>
    return %2 : tensor<f32>
  }
  // CHECK: f32=3
}