[ROCm] Disable triton gemm while running determinsim test

ROCm · Aug 21, 2024 · be97509 · be97509
1 parent ca5fb49
commit be97509
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xla/service/gpu/determinism_test.cc b/xla/service/gpu/determinism_test.cc
@@ -98,6 +98,7 @@ ENTRY e {
   if (!rocm.has_hipblaslt()) {
     GTEST_SKIP() << "No hipblas-lt support on this architecture!";
   }
+  debug_options_.set_xla_gpu_enable_triton_gemm(false);
 #endif  // TENSORFLOW_USE_ROCM
 
   debug_options_.set_xla_gpu_triton_fusion_level(0);