xla
Ignore non-XLA nodes and their direct dependents.
#6170

Merged

Ignore non-XLA nodes and their direct dependents. #6170

ysiraichi merged 5 commits into master from ysiraichi/ignore-non-xla-nodes

ysiraichi requested a review from

JackCaoG 2 years ago

ysiraichi added xla:gpu

JackCaoG added dynamo

JackCaoG commented on 2024-01-10

ysiraichi force pushed from c38c1a01 to 65cd98a5 2 years ago

Add test.

fee9fc67

Add list argument with non-XLA tensor test.

f1f0906d

Rename `FallBackNodeCollector` into `UnsupportedNodesCollector`.

416bccb7

Rename function checking tensors on XLA devices.

b99e14a9

ysiraichi force pushed from 65cd98a5 to 416bccb7 2 years ago

Add return CPU tensor test.

5b61892f

JackCaoG approved these changes on 2024-01-11

ysiraichi merged 81410785 into master 2 years ago

Reviewers

JackCaoG

Assignees

No one assigned

Labels

xla:gpu dynamo

Milestone

No milestone