pytorch/xla

Pull Requests Commits

Update fallback unit tests

Wonjoo Lee committed 2 years ago

2b59793c

Merge branch 'cpufallback' of github.com:pytorch/xla into cpufallback

seanlatias committed 2 years ago

45c14405

add call_method to fallback check list

seanlatias committed 2 years ago

cba3326d

Wonjoo Lee committed 2 years ago

f4b7407f

Refactor fallback dynamo tests

Wonjoo Lee committed 2 years ago

e5cc07d3

Revert irrelevant comments in test_dynamo.py

Wonjoo Lee committed 2 years ago

6c489c3e

Change the metric asserts back to original

Wonjoo Lee committed 2 years ago

81f26e54

Clear pending IRs after initial fetch of fallback ops

Wonjoo Lee committed 2 years ago

d70d21e6

Fix fault tolerance to fix GPU CI

Wonjoo Lee committed 2 years ago

31119bfb

Update dynamo fallback logic to support in-place operations

Wonjoo Lee committed 2 years ago

6e2cec61

wonjoo-wj committed 2 years ago

d042b8e1

Fix metrics in test_dynamo.py

Wonjoo Lee committed 2 years ago

136539de

Rename test file

seanlatias committed 2 years ago

6cf95eb6

Add more tests with numerical checks

seanlatias committed 2 years ago

028106c7

Include module into fallback checking

seanlatias committed 2 years ago

303c8e46

seanlatias committed 2 years ago

e5d243ee

seanlatias committed 2 years ago

35702f29

seanlatias committed 2 years ago

223a39f6

seanlatias committed 2 years ago

c6884136

fixed op support check

seanlatias committed 2 years ago

76cf87a1

make materialization check return true when input XLA tensors do not have XLAData or IR

seanlatias committed 2 years ago

99b5e17f

replace submodule with a function call instead

seanlatias committed 2 years ago

6f83c967

attempt to return compiled graph instead

Ubuntu committed 2 years ago

c6f09b41

move partition outside of extract compile graph

Ubuntu committed 2 years ago

205e3071

wonjoo-wj committed 2 years ago

d259ace3

Support CPU fallback for unsupported ops in dynamo

wonjoo-wj committed 2 years ago

6e75ab61

In debug mode, add debug symbols to everything but tf (#5019)

stgpetrovic committed 2 years ago

Verified 30b2aba0

Disable cmake macros (#5021)

stgpetrovic committed 2 years ago

Verified e9a93333

Expose local shards as XLAShard (#5016)

jonb377 committed 2 years ago

Verified 56960f3f

Apply sharding when creating tensors data in data loader (#5015)

jonb377 committed 2 years ago

Verified f34f0b0a

Older