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

[dynamic-shapes] Split dynamic_api_test.py

Browse Source 
PiperOrigin-RevId: 461109288
This commit is contained in:
George Necula 2022-07-14 20:18:14 -07:00 committed by jax authors
parent a35f9acbff
commit 777c129dfb
3 changed files with 1250 additions and 1205 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/BUILD
View File

@ -38,6 +38,12 @@ jax_test(
shard_count = 10,
)
jax_test(
name = "dynamic_api_test",
srcs = ["dynamic_api_test.py"],
shard_count = 2,
)
jax_test(
name = "api_util_test",
srcs = ["api_util_test.py"],

1205
tests/api_test.py
View File

File diff suppressed because it is too large Load Diff

1244
tests/dynamic_api_test.py Normal file
View File

File diff suppressed because it is too large Load Diff