pytorch/xla

Pull Requests Commits

Update xla_base tag to v0.5

yeounoh committed 2 years ago

0eb33175

Install sympy to ci base docker image

yeounoh committed 2 years ago

6197f932

Use C10_AS_INTARRAYREF_SLOW (#4325)

ezyang committed 2 years ago

Verified 1cadc304

Specified vague "Any" type definitions (#4321)

gkroiz committed 2 years ago

Verified 9794392a

Reflect that pjrt now supports `nprocs`. (#4326)

stgpetrovic committed 2 years ago

Verified 2fd2ffee

Adding support for torchrun in xla backend (#3609)

amithrm committed 2 years ago

Verified 334d1d66

[LTC] Adopt even more LazyTensor interfaces (#4317)

alanwaketan committed 2 years ago

Verified d1748e0f

[LTC] Override LazyGraphExecutor::RunPostOrder() (#4315)

alanwaketan committed 2 years ago

Verified 4ffdfe26

[LTC] Adopt more LazyGraphExecutor virtual interfaces (#4314)

alanwaketan committed 2 years ago

Verified 16e963f8

Correctly configure PJRT execution when nprocs=1. (#4285)

stgpetrovic committed 2 years ago

Verified 35a5dc12

[LTC] Inherits LazyGraphExecutor::DeviceContextArena (#4307)

alanwaketan committed 2 years ago

Verified c570f7f7

Add one more symint expand test. (#4291)

vanbasten23 committed 2 years ago

Verified 6cf15f3d

Add met.celar_all (#4310)

JackCaoG committed 2 years ago

Verified b350f363

[WIP]Disable all failed PyTorch test (#4299)

ManfeiBai committed 2 years ago

Verified e92479a0

Fixing CI (#4313)

alanwaketan committed 2 years ago

Verified 4c7c343c

[LTC] Reuse more upstream LazyGraphExecutor data structures (#4312)

alanwaketan committed 2 years ago

Verified b826def1

Revert "[LTC] Reuse more upstream LazyGraphExecutor data structures (#4303)" (#4309)

alanwaketan committed 2 years ago

Verified 07053759

Add note [Note: Re-using upstream TensorImpl] (#4284)

wonjoo-wj committed 2 years ago

Verified 385aa7c3

[LTC] Reuse more upstream LazyGraphExecutor data structures (#4303)

alanwaketan committed 2 years ago

Verified 653df694

Add a forward pass nn model with dynamism test. (#4256)

vanbasten23 committed 2 years ago

Verified 9894c96f

Update torch_xla version for colab from 1.12 to 1.13 (#4295)

vanbasten23 committed 2 years ago

Verified 7a4dcb36

Add a simple gpu doc (#4302)

JackCaoG committed 2 years ago

Verified 632de2eb

[PJRT] Update PJRT release notes with latest changes. (#4269)

will-cromar committed 2 years ago

Verified 9e2fb331

[LTC] Reuse some methods from LazyTensor (#4294)

alanwaketan committed 2 years ago

Verified 16607266

[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

Older