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

Merge pull request #25320 from ROCm:gh-9948-fix-kernel-build-upstream

Browse Source 
PiperOrigin-RevId: 704279150
This commit is contained in:
jax authors 2024-12-09 07:29:31 -08:00
commit cc258f5f61
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/tools/build_gpu_kernels_wheel.py
View File

@ -145,7 +145,7 @@ def prepare_wheel_rocm(
f"__main__/jaxlib/rocm/_linalg.{pyext}",
f"__main__/jaxlib/rocm/_prng.{pyext}",
f"__main__/jaxlib/rocm/_sparse.{pyext}",
f"__main__/jaxlib/cuda/_hybrid.{pyext}",
f"__main__/jaxlib/rocm/_hybrid.{pyext}",
f"__main__/jaxlib/rocm/_triton.{pyext}",
f"__main__/jaxlib/rocm_plugin_extension.{pyext}",
"__main__/jaxlib/version.py",