xla
[WIP] Skip AOTAutograd view-reconstruction for inference runs.
#7834

Open

[WIP] Skip AOTAutograd view-reconstruction for inference runs. #7834

ysiraichi wants to merge 8 commits into master from ysiraichi/skip-aot-view-reconstruction

Initial implementation.

76c1bcb2

Enable by default.

ae1d5d86

Fix `base` call.

a936ed32

Guard for XLA tensors.

7f92e93d

Don't store a `TracingContext`.

1d6bdba5

Fix implementation.

56a2e788

Add counter and remove prints.

73721dd3

Fix bad if.

98b57bfe

ysiraichi added xla:gpu

Reviewers

No reviews

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone