pytorch/xla

Pull Requests Commits
Update TF pin for `libtpu-nightly==0.1.dev.20221223` compatible commit (#4399)
yeounoh committed 2 years ago
Verified d43a14a9
Pjrt device attributes (#4403)
darisoy committed 2 years ago
Verified b97f05b3
Use table to represent wheels and dockers in README (#4396)
JackCaoG committed 2 years ago
Verified 73283612
Companion PR for #81762 (#3976)
lezcano committed 2 years ago
Verified 80feefaf
[LTC] Final tweaks on XLATensor and XLAGraphExecutor (#4397)
alanwaketan committed 2 years ago
Verified 1effe54a
Update outdated link to docker instructions (#4395)
JackCaoG committed 2 years ago
Verified b946e9cc
Roll back MKL related changes (#4394)
steventk-g committed 2 years ago
Verified cb1e7d4f
[PT/XLA] Fix timing of ExecuteComputation metrics. (#4327)
stgpetrovic committed 2 years ago
Verified afe7bb6b
2.0 (#4392)
JackCaoG committed 2 years ago
Verified 428e299a
Use nomkl to disable mkl optimizations (#4344)
steventk-g committed 2 years ago
Verified 40d3322e
rebase (#4380)
ManfeiBai committed 2 years ago
Verified f919b909
Make Dynamo execution also save the graph when XLA_SAVE_TENSORS_FILE is configured (#4388)
JackCaoG committed 2 years ago
Verified e0603089
Revert "Revert "Add aten::_cdist_forward"" (#4382)
JackCaoG committed 2 years ago
Verified a1d671d6
Fix a typo in test_input_output_aliases.py (#4387)
alanwaketan committed 2 years ago
Verified 6bed07c7
Add GPU doc to readme (#4389)
JackCaoG committed 2 years ago
Verified 14bdea62
Update CI test scripts to allow continuing on test failures (#4385)
wonjoo-wj committed 2 years ago
Verified cfeaedef
Update ci baseimage tag to v0.6 (#4386)
yeounoh committed 2 years ago
Verified b490eda9
Add setup for CI on TPUs (#4018)
steventk-g committed 2 years ago
Verified c4bc972b
Revert "Add aten::_cdist_forward (#4277)" (#4381)
JackCaoG committed 2 years ago
Verified fccbc6a2
Add aten::_cdist_forward (#4277)
ymwangg committed 2 years ago
Verified 3edc88f1
Improve numerical stability of torch.sigmoid (#4311)
ymwangg committed 2 years ago
Verified 453aa651
Add gpu fsdp auto_wrap test (#4378)
JackCaoG committed 2 years ago
Verified 68491026
TF pin for 12/15/2022 (#4324)
ManfeiBai committed 2 years ago
Verified 9cf2565f
Print xla_shape_ in IR too. (#4376)
alanwaketan committed 2 years ago
Verified 86bfeb61
add `auto_wrap_policy` into XLA FSDP for automatic wrapping (#4318)
ronghanghu committed 2 years ago
Verified d7d04792
Update README link to section (#4367)
JackCaoG committed 2 years ago
Verified b96c9ffe
Enable dynamo test (#4370)
JackCaoG committed 2 years ago
Verified cad848af
Enable FSDP test on gpu ci (#4372)
JackCaoG committed 2 years ago
Verified f257a203
fix type annotations in FSDP (#4371)
ronghanghu committed 2 years ago
Verified 7819db79
Add a test case for BuildInputOutputAliases (#4329)
alanwaketan committed 2 years ago
Verified 27660d79
Terms Of Use Privacy Policy Refund Policy Impressum