pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

49b15e06

alanwaketan committed 2 years ago

e02e9332

alanwaketan committed 2 years ago

536a5436

Fix some comments

alanwaketan committed 2 years ago

b9bc51f5

Fix a wrong move

alanwaketan committed 2 years ago

43ddd1bc

Adds epoch to logs

alanwaketan committed 2 years ago

1fbf62ee

Reuse _dummy_data_placeholder

alanwaketan committed 2 years ago

cf82f82b

alanwaketan committed 2 years ago

7028a239

Remove unnecessary detach

alanwaketan committed 2 years ago

c1d6df0c

Improve test_data_handle_release

alanwaketan committed 2 years ago

a38761a8

Improve the test

alanwaketan committed 2 years ago

fc458aba

Add a test case to validate that .set_ also release the data buffer

alanwaketan committed 2 years ago

6dd19ee6

alanwaketan committed 2 years ago

08d5ff78

Adds torch.no_grad for eval loops

alanwaketan committed 2 years ago

51b3c42f

alanwaketan committed 2 years ago

ce255ece

alanwaketan committed 2 years ago

2bb9e779

Replace even more .data usages with modern alternatives

alanwaketan committed 2 years ago

70e29085

alanwaketan committed 2 years ago

d15f048f

Revert "Adds .torch_pin"

alanwaketan committed 2 years ago

c2bc0b79

Replace thee last .data with .copy

alanwaketan committed 2 years ago

6719c9bc

Change to context manager

alanwaketan committed 2 years ago

587381c3

Adds .torch_pin

alanwaketan committed 2 years ago

c89c55aa

Use torch.autograd._unsafe_set_version_counter

alanwaketan committed 2 years ago

26e2f15f

Trying to replace some of .data= with .copy_()

alanwaketan committed 2 years ago

4f6e939a

[PJRT] Change the name of `LOCAL_RANK` to fix conflicts (#4646)

will-cromar committed 2 years ago

Verified 54101e76

Add spmd unit tests and linear model training to CI (#4645)

steventk-g committed 2 years ago

Verified a2b673f0

Iron out the cuda/tpuvm flag (#4657)

stgpetrovic committed 2 years ago

Verified 2d9b37b9

Revert to local nonsandboxed builds with GCC (#4651)

stgpetrovic committed 2 years ago

Verified e6c593b9

Full codegen for repeat (#4642)

cowanmeg committed 2 years ago

Verified 4ea000b9

Add experimental test coverage feature (#4316)

yeounoh committed 2 years ago

Verified a8ac0d91

Older