[XLA:GPU] Make command_buffer_thunk_test:DynamicSliceFusionCmd more t…

…olerant PiperOrigin-RevId: 726118749
openxla · Feb 12, 2025 · f995626 · f995626
1 parent 1928133
commit f995626
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/xla/backends/gpu/runtime/command_buffer_thunk_test.cc b/xla/backends/gpu/runtime/command_buffer_thunk_test.cc
@@ -25,6 +25,7 @@ limitations under the License.
 #include <variant>
 #include <vector>
 
+#include <gtest/gtest.h>
 #include "absl/status/statusor.h"
 #include "absl/strings/ascii.h"
 #include "absl/types/span.h"
@@ -1578,14 +1579,14 @@ ENTRY main.49 {
   config.set_debug_options(debug_options);
   TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module,
                           ParseAndReturnVerifiedModule(module_str, config));
-  EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1e-3, 1e-3}));
+  EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1e-3, 2e-3}));
 
   // running with module with exclusive lock on GpuExecutable
   debug_options.set_xla_gpu_require_exclusive_lock(true);
   config.set_debug_options(debug_options);
   TF_ASSERT_OK_AND_ASSIGN(module,
                           ParseAndReturnVerifiedModule(module_str, config));
-  EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1e-3, 1e-3}));
+  EXPECT_TRUE(RunAndCompare(std::move(module), ErrorSpec{1e-3, 2e-3}));
 }
 
 }  // namespace xla::gpu