diff --git a/test/ck_tile/fmha/CMakeLists.txt b/test/ck_tile/fmha/CMakeLists.txt index 60779f7f51..2656b04e57 100644 --- a/test/ck_tile/fmha/CMakeLists.txt +++ b/test/ck_tile/fmha/CMakeLists.txt @@ -7,7 +7,11 @@ set(FMHA_FWD_INSTANCES "tile_fmha_fwd_instances") set(TEST_NAME "test_ck_tile_fmha") function(add_gtest_fwd test_group) - set(V_TYPES "fp16" "bf16" "fp8bf16" "fp32") + if((GPU_TARGETS MATCHES "gfx90a" AND CK_USE_FP8_ON_UNSUPPORTED_ARCH) OR GPU_TARGETS MATCHES "gfx9[45]|gfx12") + set(V_TYPES "fp16" "bf16" "fp8bf16" "fp32") + elseif((GPU_TARGETS MATCHES "gfx90a" AND NOT CK_USE_FP8_ON_UNSUPPORTED_ARCH) OR GPU_TARGETS MATCHES "gfx11") + set(V_TYPES "fp16" "bf16" "fp32") + endif() set(CPP_TYPE_fp16 "FmhaFwdFp16") set(CPP_TYPE_bf16 "FmhaFwdBf16") set(CPP_TYPE_fp8bf16 "FmhaFwdFp8Bf16")