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

add to tests/BUILD

Browse Source
This commit is contained in:
Shuhan Ding 2024-03-12 17:17:20 -07:00
parent 4c4bcde723
commit 5a93e15bd7
Failed to extract signature
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/BUILD
View File

@ -542,6 +542,21 @@ jax_test(
] + py_deps("numpy"),
)
jax_test(
name = "lax_metal_test",
srcs = ["lax_metal_test.py"],
tags = ["notap"],
disable_backends = [
"cpu",
"gpu",
"tpu",
],
deps = [
"//jax:internal_test_util",
"//jax:lax_reference",
] + py_deps("numpy"),
)
jax_test(
name = "lax_autodiff_test",
srcs = ["lax_autodiff_test.py"],