// RUN: iree-compile \
// RUN:   %s \
// RUN:   --iree-hal-target-device=local \
// RUN:   --iree-hal-local-target-device-backends=vmvx \
// RUN:   -o=%t.vmfb && \
// RUN: iree-dump-module \
// RUN:   --output=all \
// RUN:   %t.vmfb | \
// RUN: FileCheck %s

// CHECK: @module : version 0

// CHECK: fn0
func.func @fn0(%input : tensor<f32>) -> (tensor<f32>) {
  // CHECK: ^bb0:
  %result = math.absf %input : tensor<f32>
  return %result : tensor<f32>
}

// CHECK: fn1
func.func @fn1(%input : tensor<f32>) -> (tensor<f32>) {
  // CHECK: ^bb0:
  %result = arith.mulf %input, %input : tensor<f32>
  return %result : tensor<f32>
}

// CHECK: __init