pytorch/xla

Pull Requests Commits

Merge branch 'master' into amin

miladm committed 4 years ago

db372469

Disable test_cov on TPU due to precision issue (#3032)

JackCaoG committed 4 years ago

Verified 31a26de4

rebasing to xla head to support issue #3012

miladm committed 4 years ago

9cf23956

Make kernels methods (#3012)

bdhirsh committed 4 years ago

Verified d033f8d3

adding counters to the test

miladm committed 4 years ago

4cac7d4c

addressed PR review comments

miladm committed 4 years ago

4766d1b7

initial amin implementation - not tested

miladm committed 4 years ago

87712a5c

fix transpose_ (#3018)

bdhirsh committed 4 years ago

Verified 4eedb650

Add fp16 support for ARange op (#3026)

luchangli03 committed 4 years ago

Verified 2f04ea39

Add expecttest (#3023)

JackCaoG committed 4 years ago

Verified 86998fe3

Var mean (#3014)

miladm committed 4 years ago

Verified f7351043

moved the counters out of the nested loops (#3013)

miladm committed 4 years ago

Verified 4367aa7a

Add a doc explaining recompilation. (#3015)

ailzhang committed 4 years ago

Verified ecc609e0

Std mean (#3004)

miladm committed 4 years ago

Verified 74168789

Change test order (#3011)

JackCaoG committed 4 years ago

Verified 460281be

Boxed cpu fallback (#2945)

bdhirsh committed 4 years ago

Verified cac652c3

Skip a few nll tests due to specific error messages. (#3010)

ailzhang committed 4 years ago

Verified 823e4a4c

Update official profiling tutorial link (#3008)

juneoh committed 4 years ago

Verified 9fd2eb6c

Skip a few new inplace view transpose tests. (#3007)

ailzhang committed 4 years ago

Verified 7177ce36

Use `tpu-vm scp` in on TPU VM pods (#3005)

jysohn23 committed 4 years ago

Verified 4b91c010

Update colab examples to use PyTorch/XLA 1.9. (#3000)

zcain117 committed 4 years ago

Verified e6604827

Update README for r1.9 release. (#2986)

zcain117 committed 4 years ago

Verified adc8d4de

updating tf to latest (#2992)

JackCaoG committed 4 years ago

Verified e39444e9

Add and (#2999)

JackCaoG committed 4 years ago

Verified 94a7a1a3

Set CC/CXX when running early_sync test. (#2996)

ailzhang committed 4 years ago

Verified a6b76e3e

Update pytorch-xla-profiling-colab.ipynb (#2998)

kaushikb11 committed 4 years ago

Verified af46a453

Move build commands to common.sh. (#2995)

ailzhang committed 4 years ago

Verified 2b03a256

include TPUVM doc on our README (#2994)

JackCaoG committed 4 years ago

Verified 3d76bc2e

Move xla test commands to a function. (#2993)

ailzhang committed 4 years ago

Verified 558b70bc

Nll loss nan (#2991)

JackCaoG committed 4 years ago

Verified a35a37ab

Older