PR #5000 [Dynamo] Refine CPU fallback for TD+XLA

[Dynamo] Refine CPU fallback for TD+XLA #5000

JackCaoG merged 26 commits into master from cpufallback

wonjoo-wj force pushed from b707e9d8 to a9426e9c 3 years ago

wonjoo-wj force pushed from 9cfa1d9e to 27bd4adc 3 years ago

wonjoo-wj force pushed from 9918f8c0 to 0614e543 3 years ago

wonjoo-wj marked this pull request as ready for review 3 years ago

wonjoo-wj assigned

wonjoo-wj 3 years ago

Support CPU fallback for unsupported ops in dynamo

6e75ab61

Update imports

d259ace3

move partition outside of extract compile graph

205e3071

attempt to return compiled graph instead

c6f09b41

replace submodule with a function call instead

6f83c967

make materialization check return true when input XLA tensors do not …

99b5e17f

fixed op support check

76cf87a1

fix lint issue

c6884136

add tests

223a39f6

fix cpp lint

35702f29

modify tests

e5d243ee

Include module into fallback checking

303c8e46

Add more tests with numerical checks

028106c7

Rename test file

6cf95eb6

Fix metrics in test_dynamo.py

136539de

Run linter

d042b8e1

Update dynamo fallback logic to support in-place operations

6e2cec61

Fix fault tolerance to fix GPU CI

31119bfb

wonjoo-wj commented on 2023-05-17

JackCaoG commented on 2023-05-18

Clear pending IRs after initial fetch of fallback ops

d70d21e6

Change the metric asserts back to original

81f26e54

wonjoo-wj force pushed from 0614e543 to 81f26e54 3 years ago

Revert irrelevant comments in test_dynamo.py

6c489c3e

Refactor fallback dynamo tests

e5cc07d3

Run linter

f4b7407f

add call_method to fallback check list

cba3326d

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

45c14405

JackCaoG commented on 2023-05-23

wonjoo-wj force pushed from cfd5ccd4 to ec4abf27 3 years ago

Update fallback unit tests

2b59793c

wonjoo-wj force pushed from ec4abf27 to 2b59793c 3 years ago

JackCaoG approved these changes on 2023-05-30

JackCaoG merged 3aeeb838 into master 3 years ago

Reviewers

JackCaoG

seanlatias

Assignees

wonjoo-wj

Labels

None yet

Milestone

No milestone

xla [Dynamo] Refine CPU fallback for TD+XLA #5000 Merged

[Dynamo] Refine CPU fallback for TD+XLA #5000

xla
[Dynamo] Refine CPU fallback for TD+XLA
#5000

Merged