[TEST] Reintroduce code issue that broke the build previously

I wanna confirm that the CI really doesn't pick it up. PiperOrigin-RevId: 632494435
tensorflow · Jun 26, 2024 · a5e3fa7 · a5e3fa7
1 parent 6fd879e
commit a5e3fa7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/xla/xla/service/gpu/tests/gpu_fused_mha_test.cc b/third_party/xla/xla/service/gpu/tests/gpu_fused_mha_test.cc
@@ -125,7 +125,7 @@ class MultiHeadedAttentionTest : public GpuCodegenTest {
     TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> test_module,
                             ParseAndReturnVerifiedModule(hlo_string));
     TF_ASSERT_OK_AND_ASSIGN(num_fmha_calls,
-                            CountFMHACalls(test_module->Clone()));
+                            CountFMHACalls(std::move(test_module->Clone())));
     EXPECT_EQ(num_fmha_calls, expected_num_fmha_calls);
     const Literal actual_result =
         ExecuteAndTransfer(std::move(test_module), literals);