.

openxla · Jul 30, 2024 · cec2448 · cec2448
1 parent edcc97a
commit cec2448
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/xla/pjrt/gpu/se_gpu_pjrt_compiler.cc b/xla/pjrt/gpu/se_gpu_pjrt_compiler.cc
@@ -201,15 +201,13 @@ StreamExecutorGpuCompiler::Compile(CompileOptions options,
 
 #if TENSORFLOW_USE_ROCM
 STREAM_EXECUTOR_REGISTER_MODULE_INITIALIZER(pjrt_register_se_gpu_compiler, {
-  PjRtRegisterCompiler(
-    RocmName(),
-    std::make_unique<StreamExecutorGpuCompiler>());
+  PjRtRegisterCompiler(RocmName(),
+                       std::make_unique<StreamExecutorGpuCompiler>());
 });
 #else
 STREAM_EXECUTOR_REGISTER_MODULE_INITIALIZER(pjrt_register_se_gpu_compiler, {
-  PjRtRegisterCompiler(
-    CudaName(),
-    std::make_unique<StreamExecutorGpuCompiler>());
+  PjRtRegisterCompiler(CudaName(),
+                       std::make_unique<StreamExecutorGpuCompiler>());
 });
 #endif
 }  // namespace xla