Update tests/cpp/test_persistent_buffer.cpp

Co-authored-by: Jingyue Wu <[email protected]>
NVIDIA · Feb 9, 2025 · b80bbab · b80bbab
1 parent 1d25565
commit b80bbab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cpp/test_persistent_buffer.cpp b/tests/cpp/test_persistent_buffer.cpp
@@ -1541,7 +1541,7 @@ TEST_P(LayerNormSharedMemoryTest, FusionLayerNormSharedMemoryBuffer_CUDA) {
   if (has_enough_regs_smem) {
     EXPECT_THAT(
         runtime->fusionSegments()->groups(),
-        ElementsAre(HeuristicIs(SchedulerType::InnerPersistent)));
+        UnorderedElementsAre(HeuristicIs(SchedulerType::InnerPersistent)));
     Fusion* scheduled_fusion = runtime->executors()
                                    .back()
                                    ->as<KernelExecutor>()