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

Cleanup: remove obsolete TODOs

Browse Source
This commit is contained in:
Jake VanderPlas 2020-11-20 10:15:48 -08:00
parent 8b6be20518
commit 4e738bf56b
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/lax_vmap_test.py
View File

@ -43,7 +43,6 @@ all_dtypes = jtu.dtypes.all
map, unsafe_map = safe_map, map
zip, unsafe_zip = safe_zip, zip
# TODO(jakevdp): move the following to test_util.py
compatible_shapes = [[(3,)], [(3, 4), (3, 1), (1, 4)], [(2, 3, 4), (2, 1, 4)]]
def all_bdims(*shapes):

1
tests/pmap_test.py
View File

@ -47,7 +47,6 @@ config.parse_flags_with_absl()
prev_xla_flags = None
# TODO(jakevdp): move the following to test_util.py
compatible_shapes = [[(3,)], [(3, 4), (3, 1), (1, 4)], [(2, 3, 4), (2, 1, 4)]]
def all_bdims(*shapes):