jax
203cda6f - Move test_aot_device_implicit_transfer to pjit_test.py

Commit

1 year ago

Move test_aot_device_implicit_transfer to pjit_test.py This test is not specific to compute offload and is more relevant to pjit. PiperOrigin-RevId: 680599882

References

#24011 - Move test_aot_device_implicit_transfer to pjit_test.py

Author

yashk2810

Committer

Google-ML-Automation

Parents

a046e21a

jax 203cda6f - Move test_aot_device_implicit_transfer to pjit_test.py

jax
203cda6f - Move test_aot_device_implicit_transfer to pjit_test.py