diff --git a/xla/service/gpu/dot_algorithm_support_test.cc b/xla/service/gpu/dot_algorithm_support_test.cc index f731049a8f6f6..23d21d9fbd1b5 100644 --- a/xla/service/gpu/dot_algorithm_support_test.cc +++ b/xla/service/gpu/dot_algorithm_support_test.cc @@ -228,7 +228,7 @@ INSTANTIATE_TEST_SUITE_P(DotF32ForBf16Bf16F32Tests, DotAlgorithmSupportTest, Combine(Values(PC::ALG_DOT_BF16_BF16_F32), Values(F32), Values(F32), Values(CC(8, 0)), Values(SemanticVersion{6, 0, 0}), - Values(BackendRestriction::kTritonOnly), + Values(BackendRestriction::kNoRestriction), Values(Sizes{32, 32}, Sizes{16, 2})), TestParamsToString);