[ROCm] jaxlib linalg fix

Ruturaj4 · Ruturaj4 · commit 1c75f5f9b1b9 · 2024-10-13T18:40:52.000-05:00
diff --git a/jaxlib/rocm/BUILD b/jaxlib/rocm/BUILD
@@ -273,6 +273,7 @@ cc_library(
     hdrs = ["//jaxlib/gpu:linalg_kernels.h"],
     features = ["-use_header_modules"],
     deps = [
+	":hip_blas_handle_pool",
         ":hip_gpu_kernel_helpers",
         ":hip_linalg_kernels_impl",
         ":hip_vendor",
