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

Use py_deps("absl/testing") instead of //third_party/py/absl/testing

Browse Source
This commit is contained in:
Andrey Portnoy 2024-10-09 15:24:40 -04:00
parent 62459db089
commit 2c731320af
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/pallas/BUILD
View File

@ -235,9 +235,7 @@ jax_multiplatform_test(
"//jax:pallas",
"//jax:pallas_tpu",
"//jax/_src/pallas/mosaic:random",
"//third_party/py/absl/testing:absltest",
"//third_party/py/absl/testing:parameterized",
] + py_deps("numpy"),
] + py_deps("absl/testing") + py_deps("numpy"),
)
jax_multiplatform_test(
@ -367,9 +365,7 @@ jax_multiplatform_test(
"//jax:pallas",
"//jax:pallas_tpu",
"//jax/_src/pallas/mosaic:random",
"//third_party/py/absl/testing:absltest",
"//third_party/py/absl/testing:parameterized",
] + py_deps("numpy"),
] + py_deps("absl/testing") + py_deps("numpy"),
)
jax_multiplatform_test(