diff --git a/example/65_gemm_multiply_multiply/moe_gemm1_xdl_pk_i4.cpp b/example/65_gemm_multiply_multiply/moe_gemm1_xdl_pk_i4.cpp index cd4581a9be..8a497f6905 100644 --- a/example/65_gemm_multiply_multiply/moe_gemm1_xdl_pk_i4.cpp +++ b/example/65_gemm_multiply_multiply/moe_gemm1_xdl_pk_i4.cpp @@ -402,7 +402,8 @@ int main(int argc, char* argv[]) b_element_op, cde_element_op); - if(!device_op.IsSupportedArgument(argument)) + if(!device_op.IsSupportedArgument(argument) || + !(ck::get_device_name() == "gfx942" || ck::get_device_name() == "gfx950")) { throw std::runtime_error( "wrong! device_gemm with the specified compilation parameters does " diff --git a/example/65_gemm_multiply_multiply/moe_gemm2_xdl_pk_i4.cpp b/example/65_gemm_multiply_multiply/moe_gemm2_xdl_pk_i4.cpp index f2335e9175..58f982bf46 100644 --- a/example/65_gemm_multiply_multiply/moe_gemm2_xdl_pk_i4.cpp +++ b/example/65_gemm_multiply_multiply/moe_gemm2_xdl_pk_i4.cpp @@ -409,7 +409,8 @@ int main(int argc, char* argv[]) b_element_op, cde_element_op); - if(!device_op.IsSupportedArgument(argument)) + if(!device_op.IsSupportedArgument(argument) || + !(ck::get_device_name() == "gfx942" || ck::get_device_name() == "gfx950")) { throw std::runtime_error( "wrong! device_gemm with the specified compilation parameters does "