diff --git a/tests/e2e/attention/CMakeLists.txt b/tests/e2e/attention/CMakeLists.txt index 805b9fbe0240..8ab6ca947dd1 100644 --- a/tests/e2e/attention/CMakeLists.txt +++ b/tests/e2e/attention/CMakeLists.txt @@ -78,3 +78,96 @@ iree_generated_e2e_runner_test( TARGET_CPU_FEATURES_VARIANTS "default" ) + +# To distinguish between CDNA(gfx9) and RDNA3(gfx11) +if(IREE_HIP_TEST_TARGET_CHIP MATCHES "^gfx9") + +unset(IREE_HIP_TEST_COMPILER_FLAGS) +list(APPEND IREE_HIP_TEST_COMPILER_FLAGS + "--iree-hip-target=${IREE_HIP_TEST_TARGET_CHIP}" +) + +iree_generated_e2e_runner_test( + NAME + e2e_attention_gpu_cdna3_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 + "rocm" + DRIVERS + "hip" + COMPILER_FLAGS + ${IREE_HIP_TEST_COMPILER_FLAGS} + LABELS + "noasan" + "nomsan" + "notsan" + "noubsan" + "requires-gpu-cdna3" +) + +iree_generated_e2e_runner_test( + NAME + e2e_attention_gpu_cdna3_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 + "rocm" + DRIVERS + "hip" + COMPILER_FLAGS + ${IREE_HIP_TEST_COMPILER_FLAGS} + LABELS + "noasan" + "nomsan" + "notsan" + "noubsan" + "requires-gpu-cdna3" +) + +iree_generated_e2e_runner_test( + NAME + e2e_attention_gpu_cdna3_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 + "rocm" + DRIVERS + "hip" + COMPILER_FLAGS + ${IREE_HIP_TEST_COMPILER_FLAGS} + LABELS + "noasan" + "nomsan" + "notsan" + "noubsan" + "requires-gpu-cdna3" +) +endif()