mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] jaxlib linalg fix

Browse Source
This commit is contained in:
Ruturaj4 2024-10-13 18:40:52 -05:00
parent b6f38bcc4b
commit ee223d4004
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
jaxlib/rocm/BUILD
View File

@ -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",