mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #23166 from ROCm:ci_bazel_build

Browse Source 
PiperOrigin-RevId: 665935164
This commit is contained in:
jax authors 2024-08-21 10:27:51 -07:00
commit 6a2a96c3b8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/rocm/BUILD.bazel
View File

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