pytorch/xla

Pull Requests Commits

Update CONTRIBUTING.md XRT Instructions (#4012)

alanwaketan committed 2 years ago

Verified 2885df79

Remove 2vm doc (#4058)

JackCaoG committed 2 years ago

Verified 7eff7771

Fix a divide-by-0 error for SizeDiv node. (#4051)

vanbasten23 committed 2 years ago

Verified b8268ffd

join threads before program exit (#4047)

aws-rhsoln committed 2 years ago

Verified a58ef8a3

Revert "Revert "Update amax/amin/norm signatures with int[*]? dim (#4035)"" (#4056)

wonjoo-wj committed 2 years ago

Verified ec3fd75b

Revert "Update amax/amin/norm signatures with int[*]? dim (#4035)" (#4054)

wonjoo-wj committed 2 years ago

Verified aa5d8221

Update amax/amin/norm signatures with int[*]? dim (#4035)

wonjoo-wj committed 2 years ago

Verified 2cdfa5ef

[DDP] Disable test/test_ddp.py in a GPU env (#4050)

alanwaketan committed 2 years ago

Verified 1633e032

rename TensorList -> ITensorListRef (#4013)

bdhirsh committed 2 years ago

Verified 8a8bf0b9

Minor clean up for expand_symint (#4045)

JackCaoG committed 2 years ago

Verified c2d85a00

Allow einsum to take optional parameter to align with torch (#4040)

janeyx99 committed 2 years ago

Verified f2d83a3a

Add getDynamicValue for SizeMul and SizeDiv nodes. (#4042)

vanbasten23 committed 3 years ago

Verified 23153e22

Broadcast worker ID from TPU:0 (#4043)

will-cromar committed 3 years ago

Verified 39445aa3

Remove ParamSpec from `pjrt.py` to unblock nightly build. (#4037)

will-cromar committed 3 years ago

Verified 669a2071

Add git diff to lint messaging so it is more actionable (#4041)

janeyx99 committed 3 years ago

Verified 8fff44e6

Add einsum tests (#4034)

steventk-g committed 3 years ago

Verified 5ac95b2c

Add "getDynamicValue" for SizeAdd node. (#4026)

vanbasten23 committed 3 years ago

Verified 72244be8

[DDP] Add a test case for XRT (#4014)

alanwaketan committed 3 years ago

Verified 7d6a7234

Support `expand.symint` with real symbolic symint (#3889)

miladm committed 3 years ago

Verified 31825f3d

Remove `==` constraint from `typing_extensions` in build (#4031)

will-cromar committed 3 years ago

Verified bcb212c6

Revert "Revert "Ceil forr round trunc int (#3913)" (#4019)" (#4022)

JackCaoG committed 3 years ago

Verified cd88d42d

Add Arm Compute Library (ACL) build option for torch-xla (#3542)

snadampal committed 3 years ago

Verified d3fad53b

Fix equation parsing for fallback logic (#4027)

steventk-g committed 3 years ago

Verified 5c4382de

Rename `pjrt.run_multiprocess` to `pjrt._run_multiprocess`. (#4025)

will-cromar committed 3 years ago

Verified 5bac0678

Update libtpu to align with latest TF pin update (#4030)

wonjoo-wj committed 3 years ago

Verified 4226edb6

Update TensorFlow pin (08-2022) (#3922)

wonjoo-wj committed 3 years ago

Verified 5aaab8f6

Codegen HardshrinkBackward and SoftshrinkBackward (#4002)

vanbasten23 committed 3 years ago

Verified ab15402f

Fixing a race where processes other than ones saving can remove the chkpt directory (#4007)

amithrm committed 3 years ago

Verified 0d316f3f

Cache dynamic ir result (#4008)

JackCaoG committed 3 years ago

Verified 5f45b9fe

Fix warning (#4021)

JackCaoG committed 3 years ago

Verified acbb24aa

Newer Older