diff --git a/xla/service/gpu/tests/sorting_test.cc b/xla/service/gpu/tests/sorting_test.cc index ba1468f04845b..90f1ae793d3c4 100644 --- a/xla/service/gpu/tests/sorting_test.cc +++ b/xla/service/gpu/tests/sorting_test.cc @@ -140,6 +140,18 @@ class CubSortPairsTest std::tuple, PrimitiveType, bool>> {}; TEST_P(CubSortPairsTest, SortPairs) { + // TODO(b/380814507): Remove the disabling part once fixed. + auto cc = backend() + .default_stream_executor() + ->GetDeviceDescription() + .cuda_compute_capability(); + if (cc.IsAtLeastHopper() && + std::get<0>(GetParam())->shape().element_type() == U16 && + std::get<1>(GetParam()) == F64) { + GTEST_SKIP() + << "CUB sort does not work for pair sorting (U16, F64) on Hopper."; + } + constexpr char kHloTemplate[] = R"( HloModule TestModule