From 8d42fa0b0b608273b0928bfa40a50dc99bc67206 Mon Sep 17 00:00:00 2001
From: Charles Hofer <Charles.Hofer@amd.com>
Date: Tue, 10 Dec 2024 17:38:50 -0600
Subject: [PATCH] Remove cuda include from gpu plugin extension and BUILD

---
 jaxlib/BUILD                   | 1 -
 jaxlib/gpu_plugin_extension.cc | 1 -
 2 files changed, 2 deletions(-)

diff --git a/jaxlib/BUILD b/jaxlib/BUILD
index e7ba1dd3d..875e182fa 100644
--- a/jaxlib/BUILD
+++ b/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",
diff --git a/jaxlib/gpu_plugin_extension.cc b/jaxlib/gpu_plugin_extension.cc
index ba7896aa5..9eecc1d1d 100644
--- a/jaxlib/gpu_plugin_extension.cc
+++ b/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"