@@ -1045,17 +1045,17 @@ cc_library(
1045
1045
"@xla//xla/service:hlo_proto_cc_impl" ,
1046
1046
"@com_google_absl//absl/status:statusor" ,
1047
1047
] + if_cuda ([
1048
- "@jax//jaxlib/cuda:cuda_gpu_kernels" ,
1049
- "@xla//xla/backends/profiler:profiler_backends" ,
1050
- "@xla//xla/backends/profiler/gpu:device_tracer" ,
1051
- "@xla//xla/pjrt/c:pjrt_c_api_gpu_internal" ,
1052
- "@xla//xla/service/gpu:gpu_transfer_manager" ,
1053
- "@xla//xla/service/gpu:nvptx_compiler" ,
1054
- "@xla//xla/service/gpu/model:hlo_op_profile_proto_cc_impl" ,
1055
- "@xla//xla/service/gpu/model:hlo_op_profiles" ,
1056
- "@xla//xla/stream_executor:cuda_platform" ,
1057
- "@xla//xla/stream_executor:kernel" ,
1058
- "@xla//xla/stream_executor/cuda:all_runtime" ,
1048
+ "@jax//jaxlib/cuda:cuda_gpu_kernels" ,
1049
+ "@xla//xla/backends/profiler:profiler_backends" ,
1050
+ "@xla//xla/backends/profiler/gpu:device_tracer" ,
1051
+ "@xla//xla/pjrt/c:pjrt_c_api_gpu_internal" ,
1052
+ "@xla//xla/service/gpu:gpu_transfer_manager" ,
1053
+ "@xla//xla/service/gpu:nvptx_compiler" ,
1054
+ "@xla//xla/service/gpu/model:hlo_op_profile_proto_cc_impl" ,
1055
+ "@xla//xla/service/gpu/model:hlo_op_profiles" ,
1056
+ "@xla//xla/stream_executor:cuda_platform" ,
1057
+ "@xla//xla/stream_executor:kernel" ,
1058
+ "@xla//xla/stream_executor/cuda:all_runtime" ,
1059
1059
]) + if_rocm ([
1060
1060
"@xla//xla/stream_executor:rocm_platform" ,
1061
1061
"@xla//xla/service/gpu:amdgpu_compiler" ,
0 commit comments