Antonio Kim bab1304f59 Add step closures (#84300) ...

Ports over the step closure functionality from PyTorch/XLA to Lazy Tensor Core:

References:
205ae574c0/torch_xla/core/xla_model.py (L852-L900)
205ae574c0/torch_xla/utils/closures.py (L7-L83)

CC: @wconstab @JackCaoG @Krovatkin
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84300
Approved by: https://github.com/JackCaoG, https://github.com/wconstab

2022-09-06 20:55:34 +00:00

..

__init__.py

Add step closures (#84300)

2022-09-06 20:55:34 +00:00

closure.py

Add step closures (#84300)

2022-09-06 20:55:34 +00:00

computation.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00

config.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00

debug.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00

device_context.py

Add step closures (#84300)

2022-09-06 20:55:34 +00:00

extract_compiled_graph.py

Fix use-dict-literal lint (#83718)

2022-08-24 00:26:46 +00:00

ir_cache.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00

metrics.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00

tensor_factory_functions.py

Dynamo+LTC: merging related code from staging branch to master (#75046)

2022-04-02 00:23:15 +00:00

ts_backend.py

Apply ufmt to torch internal (#81643)

2022-07-22 02:19:50 +00:00