pytorch/xla

Pull Requests Commits

Remove .torch_pin

alanwaketan committed 2 years ago

335df211

alanwaketan committed 2 years ago

b55aec84

alanwaketan committed 2 years ago

ccf493e2

Replace TORCH_CHECK with XLA_CHECK

alanwaketan committed 2 years ago

864b2e70

alanwaketan committed 2 years ago

15d98711

alanwaketan committed 2 years ago

d5afdb79

Remove GetNextTensorId

alanwaketan committed 2 years ago

50707603

Replace data_ptr() with data()

alanwaketan committed 2 years ago

bed594ba

Change data() to return shared_ptr

alanwaketan committed 2 years ago

d6feb445

Rename CurrentXlaData to CurrentDataHandle

alanwaketan committed 2 years ago

95ef5069

Make data() virtual

alanwaketan committed 2 years ago

fe5e4e22

Remove somee duplicated methods as the base class

alanwaketan committed 2 years ago

af146d77

Let XLATensor::Data inherits torch::lazy::LazyTensor::Data

alanwaketan committed 2 years ago

cf872d94

Rename Data.xla_data to handle

alanwaketan committed 2 years ago

0ffae4a5

[LTC] Inherits LazyGraphExecutor (#4296)

alanwaketan committed 2 years ago

Verified 036728e9

api to grad base seed as device data (#4293)

shunting314 committed 2 years ago

Verified feba771c

add docs readme back (#4301)

JackCaoG committed 2 years ago

Verified f80a9814

Document environment variables (#4273)

steventk-g committed 2 years ago

Verified 36c53d63

Configure GCB triggers with Terraform (#4267)

will-cromar committed 2 years ago

Verified c9f4c353

Add virtual device metrics (#4279)

steventk-g committed 2 years ago

Verified e4151962

Sharding annotation for empty node (#4292)

yeounoh committed 2 years ago

Verified 5eb9ad46

Update documentation for op related guides (#4283)

wonjoo-wj committed 2 years ago

Verified a1c054da

Lower _native_batch_norm_legit (#4282)

JackCaoG committed 2 years ago

Verified 6fcc8cfc

Fix tf_patches/f16_abi_clang.diff ill-format. (#4278)

vanbasten23 committed 2 years ago

Verified 6a46831e

[PJRT] Create function to spawn threads instead of processes. (#4263)

will-cromar committed 2 years ago

Verified 5cefe63a

Introduce XLAGraphExecutor (#4270)

alanwaketan committed 2 years ago

Verified c20bcdd3

[LTC] Naively inherits LazyTensor (#4271)

alanwaketan committed 2 years ago

Verified f2c87eec

Enable dynamic shape tests on TPUVM (#4259)

JackCaoG committed 2 years ago

Verified 1685a288

Move tensor methods out from XLATensor (#4260)

alanwaketan committed 2 years ago

Verified 4fd09cc8

Clean up some XLATensor unused interfaces (#4257)

alanwaketan committed 2 years ago

Verified 965ca9a2

Older