tpu amp
0df47439
Add torch pin
7997da0f
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
6ac8262d
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
042a6311
updates
401eaf61
Clean up
041fce29
Delete test_fsdp_auto_wrap_amp.py
0aa5011d
updates
2157e2c0
Update autocast key to pick between Cuda and Xla. Unit tests
e247b43c
lint
7d215d1b
moving code from pt to ptxla
661759ad
fixes
e339506c
lint
03da7ec7
move autocast+test from common.sh to run_tests.sh
e4a0daae
updates
ff8f10bb
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
a0437a5e
updates with pytorch
410912eb
lint
aabbd128
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
1ce912ca
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
dfe1f276
build autocast_mode
52753340
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
0652dd67
Merge branch 'master' of https://github.com/cowanmeg/xla into amp
089f2961
Merge branch 'master' of https://github.com/pytorch/xla into amp
e1243d81
Merge branch 'master' of https://github.com/pytorch/xla into amp
85d607a6
Merge branch 'master' of https://github.com/pytorch/xla into amp
9241b0a3
Disable bazel remote cache
643691c1
experiment with no new files
4eba81f7
revert back
ffac43f5
lint
4097324d
JackCaoG
approved these changes
on 2023-06-09
Login to write a write a comment.
Login via GitHub