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()); + PjRtRegisterCompiler(RocmName(), + std::make_unique()); }); #else STREAM_EXECUTOR_REGISTER_MODULE_INITIALIZER(pjrt_register_se_gpu_compiler, { - PjRtRegisterCompiler( - CudaName(), - std::make_unique()); + PjRtRegisterCompiler(CudaName(), + std::make_unique()); }); #endif } // namespace xla