Merge pull request jax-ml#25407 from ROCm:remove-cuda-import-in-plugin-upstream

PiperOrigin-RevId: 705168796

Author: Google-ML-Automation

jaxlib/BUILD

@@ -222,7 +222,6 @@ cc_library(
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings:str_format",
-        "@local_config_cuda//cuda:cuda_headers",
         "@nanobind",
         "@tsl//tsl/platform:statusor",
         "@xla//xla:util",

jaxlib/gpu_plugin_extension.cc

@@ -22,7 +22,6 @@ limitations under the License.
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
 #include "absl/strings/str_format.h"
-#include "third_party/gpus/cuda/include/cuda.h"
 #include "jaxlib/kernel_nanobind_helpers.h"
 #include "xla/ffi/api/c_api.h"
 #include "xla/pjrt/c/pjrt_c_api.h"