pytorch/xla

Pull Requests Commits

Make computation to take string when used for compilation

JackCaoG committed 3 years ago

649f279f

add RegisterXLAXLANativeFunctions

JackCaoG committed 3 years ago

2f78eb80

Add missing namespace

JackCaoG committed 3 years ago

093aa7b6

JackCaoG committed 3 years ago

7ec691c5

Get rid of the old code gen file

JackCaoG committed 3 years ago

61b92371

JackCaoG committed 3 years ago

05dca761

Register XLA lazy backend

JackCaoG committed 3 years ago

80c86f21

Remove l1 loss backward (#3624)

JackCaoG committed 3 years ago

Verified de45c7c5

Revert "[TEST] test XLA CI w/ SymInts (#3639)" (#3651)

miladm committed 3 years ago

Verified 40e69b79

[TEST] test XLA CI w/ SymInts (#3639)

Krovatkin committed 3 years ago

Verified ab2742fc

Revert "Update cpp test"

JackCaoG committed 3 years ago

d2d03ad2

Update cpp test

JackCaoG committed 3 years ago

9b674c20

Implement the remaining backend interface (#3627)

JackCaoG committed 3 years ago

Verified e2235ff4

Use tcmalloc on TPUVM docker (#3647)

JackCaoG committed 3 years ago

Verified a5d83f41

Load libtpu in XRT local service (#3645)

will-cromar committed 3 years ago

Verified fe5f2db8

code for async scalar transfer (#3618)

aws-rhsoln committed 3 years ago

Verified 01e3c964

Cleanup README (#3634)

JackCaoG committed 3 years ago

Verified 4650c0ec

Fix ScopePusher regression issue (#3637)

wonjoo-wj committed 3 years ago

Verified f72dcc65

Install libtpu on TPUVM docker (#3635)

JackCaoG committed 3 years ago

Verified 54b4ddcf

fix torchvision pinning (#3643)

bdhirsh committed 3 years ago

Verified 1f9d52c7

Revert "Revert "Index device error (#3632)" (#3640)" (#3641)

bdhirsh committed 3 years ago

Verified 7ae5f50f

Revert "Index device error (#3632)" (#3640)

JackCaoG committed 3 years ago

Verified 7c99ad06

Index device error (#3632)

bdhirsh committed 3 years ago

Verified 9d5f6f68

delegate sym_sizes to sizes since Pytorch is now using sym_sizes (#3636)

Krovatkin committed 3 years ago

Verified 4f9d3bdc

replace out= lowerings with functional lowerings (#3620)

bdhirsh committed 3 years ago

Verified 9d8706f6

Build nightly torchaudio wheel (#3626)

JackCaoG committed 3 years ago

Verified 943f38f8

Refactor PjRt tests to take advantage of dynamic libtpu loading. (#3628)

will-cromar committed 3 years ago

Verified 3b2bbc8f

More conservative IsSparseGather check to favor dense gather on TPU (#3629)

yeounoh committed 3 years ago

Verified 91095239

Add TPUVM wheel

JackCaoG committed 3 years ago

0c8a82d6

Update nightly version (#3621)

JackCaoG committed 3 years ago

Verified 8ae2ec50

Older