From cec244808a8df163f9a803db450ca2bebdda9315 Mon Sep 17 00:00:00 2001
From: eaplatanios <e.a.platanios@gmail.com>
Date: Mon, 29 Jul 2024 20:38:23 -0700
Subject: [PATCH] .

---
 xla/pjrt/gpu/se_gpu_pjrt_compiler.cc | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/xla/pjrt/gpu/se_gpu_pjrt_compiler.cc b/xla/pjrt/gpu/se_gpu_pjrt_compiler.cc
index ebb125b18a6a1..ea9541ce8a03b 100644
--- 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