mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46: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"], hdrs = ["//jaxlib/gpu:linalg_kernels.h"],
features = ["-use_header_modules"], features = ["-use_header_modules"],
deps = [ deps = [
":hip_blas_handle_pool",
":hip_gpu_kernel_helpers", ":hip_gpu_kernel_helpers",
":hip_linalg_kernels_impl", ":hip_linalg_kernels_impl",
":hip_vendor", ":hip_vendor",