| # TODO: (#17751) Add the arm_64 tests when the bug resolved. See: |
| # https://github.com/iree-org/iree/actions/runs/10468944505/job/28990909321#step:4:9815 |
| if(IREE_ARCH STREQUAL "arm_64") |
| return() |
| endif() |
| |
| iree_generated_e2e_runner_test( |
| NAME |
| e2e_attention_cpu_f16_f16_f16_small |
| TEST_TYPE |
| attention |
| GENERATOR |
| "generate_e2e_attention_tests.py" |
| GENERATOR_ARGS |
| "--query_type=f16" |
| "--key_type=f16" |
| "--value_type=f16" |
| "--shapes=small" |
| TEST_RUNNER |
| iree_tools_testing_e2e_iree-e2e-attention-test |
| TARGET_BACKENDS |
| "llvm-cpu" |
| DRIVERS |
| "local-task" |
| LABELS |
| "hostonly" |
| "local" |
| TARGET_CPU_FEATURES_VARIANTS |
| "default" |
| ) |
| |
| iree_generated_e2e_runner_test( |
| NAME |
| e2e_attention_cpu_f16_f16_f16_medium |
| TEST_TYPE |
| attention |
| GENERATOR |
| "generate_e2e_attention_tests.py" |
| GENERATOR_ARGS |
| "--query_type=f16" |
| "--key_type=f16" |
| "--value_type=f16" |
| "--shapes=medium" |
| TEST_RUNNER |
| iree_tools_testing_e2e_iree-e2e-attention-test |
| TARGET_BACKENDS |
| "llvm-cpu" |
| DRIVERS |
| "local-task" |
| LABELS |
| "hostonly" |
| "local" |
| TARGET_CPU_FEATURES_VARIANTS |
| "default" |
| ) |
| |
| iree_generated_e2e_runner_test( |
| NAME |
| e2e_attention_cpu_f16_f16_f16_large |
| TEST_TYPE |
| attention |
| GENERATOR |
| "generate_e2e_attention_tests.py" |
| GENERATOR_ARGS |
| "--query_type=f16" |
| "--key_type=f16" |
| "--value_type=f16" |
| "--shapes=large" |
| TEST_RUNNER |
| iree_tools_testing_e2e_iree-e2e-attention-test |
| TARGET_BACKENDS |
| "llvm-cpu" |
| DRIVERS |
| "local-task" |
| LABELS |
| "hostonly" |
| "local" |
| TARGET_CPU_FEATURES_VARIANTS |
| "default" |
| ) |
