jax
aaac1d8f
- Skip `test_spmd_preserves_input_sharding_vmap_grad` unless `xla_extension_version >= 258`
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Skip `test_spmd_preserves_input_sharding_vmap_grad` unless `xla_extension_version >= 258` PiperOrigin-RevId: 626062015
References
#20815 - Skip `test_spmd_preserves_input_sharding_vmap_grad` unless `xla_extension_version >= 258`
Author
yashk2810
Committer
a-googler
Parents
51763d8b
Loading