explicitly set xla_gpu_simplify_all_fp_conversions=true

openxla · Jun 7, 2023 · 5c853fa · 5c853fa
1 parent d5f6fb1
commit 5c853fa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xla/service/gpu/tests/gemm_rewrite_test.cc b/xla/service/gpu/tests/gemm_rewrite_test.cc
@@ -1275,6 +1275,7 @@ class LegacyCublasGemmRewriteTest : public GemmRewriteTest {
   DebugOptions GetDebugOptionsForTest() override {
     DebugOptions debug_options = GemmRewriteTest::GetDebugOptionsForTest();
     debug_options.set_xla_gpu_enable_cublaslt(false);
+    debug_options.set_xla_gpu_simplify_all_fp_conversions(true);
     return debug_options;
   }
 };
@@ -1951,6 +1952,7 @@ class CublasLtGemmRewriteTest : public GemmRewriteTest {
   DebugOptions GetDebugOptionsForTest() override {
     DebugOptions debug_options = GemmRewriteTest::GetDebugOptionsForTest();
     debug_options.set_xla_gpu_enable_cublaslt(true);
+    debug_options.set_xla_gpu_simplify_all_fp_conversions(true);
     return debug_options;
   }
 };