From ee223d40045823279812ccdb1cd1d82a8e0883db Mon Sep 17 00:00:00 2001 From: Ruturaj4 Date: Sun, 13 Oct 2024 18:40:52 -0500 Subject: [PATCH] [ROCm] jaxlib linalg fix --- jaxlib/rocm/BUILD | 1 + 1 file changed, 1 insertion(+) diff --git a/jaxlib/rocm/BUILD b/jaxlib/rocm/BUILD index 598741522..71aab99e7 100644 --- a/jaxlib/rocm/BUILD +++ b/jaxlib/rocm/BUILD @@ -289,6 +289,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",