pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

bbb7e582

alanwaketan committed 2 years ago

dada0e1a

Adds more comments

alanwaketan committed 2 years ago

2ca9d596

Add one more comment

alanwaketan committed 2 years ago

8eb9179f

Introduce a new XLATensor::Create

alanwaketan committed 2 years ago

7bb9e711

Adds some comments

alanwaketan committed 2 years ago

ec74ed7e

Use nomkl to disable mkl optimizations (#4344)

steventk-g committed 2 years ago

Verified 40d3322e

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

Update test_xla_sharding.py (#4364)

wonjoo-wj committed 2 years ago

Verified 0d89710a

Fix warnings in `xrt_init.py` (#4341)

will-cromar committed 2 years ago

Verified 7881f536

Add unsharded computation test for virtual device (#4215)

steventk-g committed 2 years ago

Verified 43116740

Older