tests/e2e/stablehlo_models/generated_e2e_model_tests.cmake

################################################################################
# Autogenerated by build_tools/testing/generate_cmake_e2e_model_tests.py       #
# To update the tests, modify definitions in the generator and regenerate this #
# file.                                                                        #
################################################################################

iree_benchmark_suite_module_test(
  NAME "mobilenet_v1_fp32_correctness_test"
  DRIVER "local-sync"
  EXPECTED_OUTPUT "mobilenet_v1_fp32_expected_output.txt"
  MODULES
    "riscv_64-Linux=iree_module_MobileNetV1_fp32_tflite___riscv_64-generic-linux_gnu-llvm_cpu__default-flags_/module.vmfb"
    "x86_64-Linux=iree_module_MobileNetV1_fp32_tflite___x86_64-cascadelake-linux_gnu-llvm_cpu__default-flags_dt-uk_/module.vmfb"
  RUNNER_ARGS
    "--function=main"
    "--device_allocator=caching"
    "--input=1x224x224x3xf32=0"
)

iree_benchmark_suite_module_test(
  NAME "efficientnet_int8_correctness_test"
  DRIVER "local-sync"
  EXPECTED_OUTPUT "efficientnet_int8_expected_output.txt"
  MODULES
    "x86_64-Linux=iree_module_EfficientNet_int8_tflite___x86_64-cascadelake-linux_gnu-llvm_cpu__default-flags_dt-uk_/module.vmfb"
  RUNNER_ARGS
    "--function=main"
    "--device_allocator=caching"
    "--input=1x224x224x3xui8=0"
)

iree_benchmark_suite_module_test(
  NAME "deeplab_v3_fp32_correctness_test"
  DRIVER "local-sync"
  EXPECTED_OUTPUT "https://storage.googleapis.com/iree-model-artifacts/deeplab_v3_fp32_input_0_expected_output.npy"
  MODULES
    "arm_64-Android=iree_module_DeepLabV3_fp32_tflite___armv8.2-a-generic-linux_android29-llvm_cpu__default-flags_dt-uk_/module.vmfb"
    "x86_64-Linux=iree_module_DeepLabV3_fp32_tflite___x86_64-cascadelake-linux_gnu-llvm_cpu__default-flags_dt-uk_/module.vmfb"
  RUNNER_ARGS
    "--function=main"
    "--device_allocator=caching"
    "--expected_f32_threshold=0.001"
    "--input=1x257x257x3xf32=0"
)

iree_benchmark_suite_module_test(
  NAME "person_detect_int8_correctness_test"
  DRIVER "local-sync"
  EXPECTED_OUTPUT "1x2xi8=[72 -72]"
  MODULES
    "riscv_32-Linux=iree_module_PersonDetect_int8_tflite___riscv_32-generic-linux_gnu-llvm_cpu__default-flags_/module.vmfb"
    "riscv_64-Linux=iree_module_PersonDetect_int8_tflite___riscv_64-generic-linux_gnu-llvm_cpu__default-flags_/module.vmfb"
    "x86_64-Linux=iree_module_PersonDetect_int8_tflite___x86_64-cascadelake-linux_gnu-llvm_cpu__default-flags_dt-uk_/module.vmfb"
  RUNNER_ARGS
    "--function=main"
    "--device_allocator=caching"
    "--input=1x96x96x1xi8=0"
)