pytorch/xla

Pull Requests Commits
Fix profiler
alanwaketan committed 1 year ago
278e6fe3
enable spmd
alanwaketan committed 1 year ago
98bfb098
Enable backward
alanwaketan committed 1 year ago
42149014
Optional backward
alanwaketan committed 1 year ago
1f22690a
Improve a bit
alanwaketan committed 1 year ago
a0bff127
make backward a single graph
alanwaketan committed 1 year ago
d60e201d
tmp
alanwaketan committed 1 year ago
0ef1ff0e
Fix linters
alanwaketan committed 1 year ago
51f51a07
Improve test case
alanwaketan committed 1 year ago
401ca633
Enable backward
alanwaketan committed 1 year ago
899735f8
Support forward
alanwaketan committed 1 year ago
5a89157b
initial commit
alanwaketan committed 1 year ago
9bf7c12d
Add an e2e test
alanwaketan committed 1 year ago
01a5f871
[SPMD] Introduce high level manual sharding APIs (#6931)
alanwaketan committed 1 year ago
Verified 9b2ac4b1
Add examples for training (#6929)
qihqi committed 1 year ago
Verified 781ee938
Add CONTRIBUTING.md (#6924)
qihqi committed 1 year ago
Verified aee1d37f
[SPMD] Support SPMDShardToFullShape (#6925)
alanwaketan committed 1 year ago
Verified 6e34bbed
Update CUDA 12.1 dev image to python 3.10 (#6928)
will-cromar committed 1 year ago
Verified 42b028e2
[SPMD] Fix test_spmd_full_to_shard_shape for v2 (#6927)
alanwaketan committed 1 year ago
Verified dd726d3f
Add support for few basic mutation ops (#6895)
qihqi committed 1 year ago
Verified dc17f9c0
[SPMD] Support SPMDFullToShardShape (#6922)
alanwaketan committed 1 year ago
Verified 2763248f
Add a devcontainer for gpu dev (#6923)
JackCaoG committed 1 year ago
Verified a7a1357d
Add dynamo expand test. (#6659)
ysiraichi committed 1 year ago
Verified 58a412cb
Build plugin during nightly build (#6917)
will-cromar committed 1 year ago
Verified 5a269d7b
Install libtpu for release dockers (#6919)
alanwaketan committed 1 year ago
Verified ec5615c9
[SPMD] Support manual sharding (#6915)
alanwaketan committed 1 year ago
Verified e5513ff4
fix pixel_shuffle return empty (#6907)
zpcore committed 1 year ago
Verified 43f10715
Build TPU package in CI (#6918)
will-cromar committed 1 year ago
Verified 7375a8ad
Add an option to not generate weights. (#6909)
qihqi committed 1 year ago
Verified fc11bbbf
Add `xla.sync` (#6914)
will-cromar committed 1 year ago
Verified 42452717
Terms Of Use Privacy Policy Refund Policy Impressum