xla
Add tests for DTensor.from_local() support for XLA device mesh
#9504
Open
Overview
Commits
5
Changes
View On GitHub

Add tests for DTensor.from_local() support for XLA device mesh #9504

Hoomaaan wants to merge 5 commits into pytorch:master from Hoomaaan:fromLocal_wspec
Hoomaaan
aws-cph Implement XLAShardedTensor._spec and test
9af66d89
aws-cph Removed auto wrapping sharding propagation, added cached spec invalid…
6c2de4d9
aws-cph Removing lazy import
ef5f9ad0
aws-cph Added test for catching thrown error in spec
b395d4cd
Hoomaaan Test for Routing XLA device handling through distribute_tensor to ens…
7e8003f1
Hoomaaan Hoomaaan force pushed from 5efeb100 to 7e8003f1 147 days ago
qihqi qihqi requested a review from bfolie bfolie 99 days ago
qihqi qihqi requested a review from rpsilva-aws rpsilva-aws 99 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum