pytorch/torch/jit
Elias Ellison f8a2efc190 Make fusion strategy api public (#72639)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72639

Test Plan: Imported from OSS

Reviewed By: soulitzer

Differential Revision: D34159123

Pulled By: eellison

fbshipit-source-id: 27e4d9694a83e8d6829009882715be4308c96a9f
(cherry picked from commit 1cadcd2f75)
2022-02-16 03:45:15 +00:00
..
_passes Move the apply_tensor_props into its own function for more public use (#67786) 2022-01-10 11:26:03 -08:00
mobile Back out "[pytorch][PR] Add ability for a mobile::Module to save as flatbuffer" (#69796) 2021-12-10 21:29:53 -08:00
__init__.py Make fusion strategy api public (#72639) 2022-02-16 03:45:15 +00:00
_async.py
_builtins.py Update min python version to 3.7 in setup.py and mypy configs (#71494) 2022-01-20 00:03:57 +00:00
_check.py Fix comment indentation issue (#70227) 2022-01-06 19:14:39 -08:00
_freeze.py [JIT] optimize_for_inference on methods other than forward (#69367) 2021-12-07 12:36:47 -08:00
_fuser.py Make fusion strategy api public (#72639) 2022-02-16 03:45:15 +00:00
_logging.py
_monkeytype_config.py
_pickle.py
_recursive.py Reland torch.ops API change machinery with the core functionality disabled (#71785) 2022-02-02 16:06:29 +00:00
_script.py Nvfuser code bump 2_1_2022 (#72127) 2022-02-15 00:43:16 +00:00
_serialization.py Bump version number to 7 and compile old operators with old schema (#68358) 2022-01-05 23:57:22 -08:00
_state.py
_trace.py refactor assert_close to be more modular (#67794) 2021-11-19 12:37:16 -08:00
annotations.py Reland torch.ops API change machinery with the core functionality disabled (#71785) 2022-02-02 16:06:29 +00:00
frontend.py Disallow annotations on instance attributes outside __init__ (#67051) 2021-10-25 16:20:47 -07:00
generate_bytecode.py Move bytecode generation to python (#71681) 2022-01-28 02:33:00 +00:00
quantized.py Deprecate torch.jit.quantized (#72690) 2022-02-11 23:18:28 +00:00
supported_ops.py
unsupported_tensor_ops.py