pytorch/torch/_dynamo
Will Constable 2dca418112 Reland basic dynamo support for traceable collectives (#100476)
Relative to the original land, this also contains:
- Fix torchdeploy import of functional collectives
- Can't import torchdynamo utils due to torch._refs being missing

Pull Request resolved: https://github.com/pytorch/pytorch/pull/100476
Approved by: https://github.com/kumpera
2023-05-04 04:25:35 +00:00
..
backends Use Boxed Calling Convention for AOT Eager (#100417) 2023-05-04 01:22:36 +00:00
repro Improve minifier printing to be more chatty when it makes sense (#100486) 2023-05-04 02:51:26 +00:00
variables Fix graph break in a common func(self, *args) pattern (Faster stable diffusion) (#100444) 2023-05-04 03:38:52 +00:00
__init__.py Allow in graph einops operators (#99631) 2023-04-21 03:14:38 +00:00
allowed_functions.py Reland basic dynamo support for traceable collectives (#100476) 2023-05-04 04:25:35 +00:00
bytecode_analysis.py [dynamo 3.11] implement 3.11 exceptiontable (#96511) 2023-04-18 07:53:24 +00:00
bytecode_transformation.py [dynamo 3.11] implement 3.11 exceptiontable (#96511) 2023-04-18 07:53:24 +00:00
codegen.py [dynamo][numpy] Add NumpyTensorVariable to translate ndarray attribute calls to tensor attributes (#95849) 2023-04-27 16:18:35 +00:00
comptime.py Add a convenience shortcut for accessing size on ComptimeVar (#95404) 2023-02-27 02:02:02 +00:00
config_utils.py Do not use pickle to output config entries in repro scripts (#100354) 2023-05-02 11:44:01 +00:00
config.py Reland basic dynamo support for traceable collectives (#100476) 2023-05-04 04:25:35 +00:00
convert_frame.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
debug_utils.py Misc accuracy improvements on minifier (#100447) 2023-05-04 02:51:26 +00:00
eval_frame.py Update docstring for dynamo.export tracing_mode (#100205) 2023-04-29 02:12:08 +00:00
exc.py Improved message to suppress errors in _dynamo/exc.py (#97345) 2023-04-28 01:12:08 +00:00
external_utils.py Fix torch.compile() on a skipped module (#98894) 2023-04-22 16:10:55 +00:00
guards.py Remove special casing for stride/size setup for guards (#100456) 2023-05-03 01:59:52 +00:00
hooks.py Add missing mypy-nofollow.ini (#90179) 2022-12-08 01:05:12 +00:00
logging.py Fix flake8 lint errors - part 2 - manual fixes (#99799) 2023-04-24 06:03:26 +00:00
mutation_guard.py [Dynamo] Fix constructing lazy submodule inside of lazy module's initialize_parameters (#100047) 2023-04-26 23:36:31 +00:00
output_graph.py Partition modules (#98628) 2023-05-03 23:31:56 +00:00
profiler.py Revert "Replace _dynamo.config with an object instead of module (#96455)" 2023-04-12 15:06:14 +00:00
replay_record.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
resume_execution.py [dynamo 3.11] implement 3.11 exceptiontable (#96511) 2023-04-18 07:53:24 +00:00
side_effects.py Ensure optimizer state references are cleared (#100282) 2023-05-01 22:25:07 +00:00
skipfiles.py [hotfix] Do not import torch.ao.quantization._pt2e from dynamo (#100194) 2023-04-28 01:32:23 +00:00
source.py Pass locals to enum_repr to correctly make the guard str for enums (#99680) 2023-04-21 07:14:49 +00:00
symbolic_convert.py Activate TracingContext earlier (#100043) 2023-05-03 20:55:30 +00:00
test_case.py Revert "Replace _dynamo.config with an object instead of module (#96455)" 2023-04-12 15:06:14 +00:00
test_minifier_common.py Misc accuracy improvements on minifier (#100447) 2023-05-04 02:51:26 +00:00
testing.py [dynamo][numpy] Add NumpyTensorVariable to translate ndarray attribute calls to tensor attributes (#95849) 2023-04-27 16:18:35 +00:00
types.py Test and fix guard fail message in CompileProfiler (#97055) 2023-03-22 02:17:57 +00:00
utils.py Misc accuracy improvements on minifier (#100447) 2023-05-04 02:51:26 +00:00