From c41d644886674fe868b3d28b55e0986b848a2a0b Mon Sep 17 00:00:00 2001 From: Ruturaj4 Date: Tue, 20 Aug 2024 23:59:40 -0500 Subject: [PATCH] [ROCm] Fix bazel build issue --- jaxlib/rocm/BUILD.bazel | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/jaxlib/rocm/BUILD.bazel b/jaxlib/rocm/BUILD.bazel index ce733d827..0c1fe2582 100644 --- a/jaxlib/rocm/BUILD.bazel +++ b/jaxlib/rocm/BUILD.bazel @@ -60,8 +60,8 @@ cc_library( rocm_library( name = "hip_make_batch_pointers", - srcs = ["//third_party/py/jax/jaxlib/gpu:make_batch_pointers.cu.cc"], - hdrs = ["//third_party/py/jax/jaxlib/gpu:make_batch_pointers.h"], + srcs = ["//jaxlib/gpu:make_batch_pointers.cu.cc"], + hdrs = ["//jaxlib/gpu:make_batch_pointers.h"], deps = [ ":hip_vendor", "@local_config_rocm//rocm:rocm_headers",