pytorch/torch/_dynamo
2024-03-07 22:25:43 +00:00
..
backends Pass inductor strides forward in ddp optimizer (#120523) 2024-02-29 22:25:00 +00:00
repro Add missing newline to repro and some utility thing in repro (#121051) 2024-03-04 22:52:54 +00:00
variables Subclass view fake-ification via reified ViewFuncs (#118405) 2024-03-07 19:56:16 +00:00
__init__.py [torch.compile] Provide capability to register callback on compile start/stop (#120764) 2024-02-29 07:37:52 +00:00
_trace_wrapped_higher_order_op.py [Compiled Autograd] Introduce BackwardState capture (#120382) 2024-02-28 20:36:47 +00:00
bytecode_analysis.py [dynamo] Enable typechecking for bytecode_analysis.py (#112972) 2023-11-08 21:17:45 +00:00
bytecode_transformation.py [BE]: FURB187 Use inplace reverse on lists: faster, more readable. (#121140) 2024-03-05 01:36:17 +00:00
cache_size.py Chore: improve log message about cache size limit exceeded (#116557) 2024-01-17 06:07:18 +00:00
callback.py [torch.compile] Provide capability to register callback on compile start/stop (#120764) 2024-02-29 07:37:52 +00:00
code_context.py
codegen.py [dynamo] Add some helpers to PyCodegen (#120684) 2024-02-27 18:46:51 +00:00
compiled_autograd.py Support higher order op functionalization in predispatch IR (#115314) 2024-03-01 09:13:47 +00:00
comptime.py Make torch._dynamo.mark_static work inside graph (#118962) 2024-02-02 20:01:27 +00:00
config.py Revert "Limit loop unrolling (#120023)" 2024-03-02 00:04:08 +00:00
convert_frame.py [torch.compile] Provide capability to register callback on compile start/stop (#120764) 2024-02-29 07:37:52 +00:00
current_scope_id.py
debug_utils.py [inductor] When generating debug logs don't fail if nvcc not found (#120346) 2024-02-22 14:25:34 +00:00
decorators.py [Dynamo][15/N] Merge allow_in_graph/inline/skip trace rules check into trace_rule.lookup (#118971) 2024-02-07 05:15:39 +00:00
device_interface.py Reconstruct streams via global registration, temporary impl to unblock FSDP (#117386) 2024-01-13 07:03:33 +00:00
eval_frame.py fix dupe deprecated warning in dynamo export (#120896) 2024-02-29 18:57:42 +00:00
exc.py [exportdb] Remove torch/fb/exportdb (#117866) 2024-01-22 17:41:33 +00:00
external_utils.py [dynamo] Support module backwards hooks (#120685) 2024-03-01 02:24:26 +00:00
funcname_cache.py
guards.py [dynamo] Remove extra guard for tensor constant attrs (#121106) 2024-03-05 17:16:04 +00:00
hooks.py [dynamo] Enable typechecking for hooks.py (#112565) 2023-11-04 19:37:06 +00:00
logging.py Add stack trace to "start tracing" log (#118217) 2024-01-25 06:53:12 +00:00
mutation_guard.py [dynamo] Make {mutation_guard,symbolic_convert,side_effects}.py pass follow_imports typechecking (#113610) 2023-11-16 01:54:00 +00:00
output_graph.py [dynamo] relax missing symbols runtime assert (#121339) 2024-03-07 14:53:38 +00:00
polyfill.py [dynamo] Desugar accumulate_grad, fix .grad handling (#120590) 2024-02-27 10:12:26 +00:00
profiler.py Remove size asserts from fx_insert_profiling (#114830) 2023-12-04 19:08:36 +00:00
replay_record.py [CI] Install dill in ci (#116214) 2024-01-24 23:42:35 +00:00
resume_execution.py Add torch_dynamo to resume_in for ease of debugging (#118201) 2024-01-25 06:52:17 +00:00
side_effects.py [dynamo] Support module backwards hooks (#120685) 2024-03-01 02:24:26 +00:00
source.py Introduce EphemeralSource for symbols that should be simplified out (#120948) 2024-03-06 02:30:52 +00:00
symbolic_convert.py Revert "Limit loop unrolling (#120023)" 2024-03-02 00:04:08 +00:00
test_case.py [mypy] declare type for DynamoTestCase._exit_stack (#119084) 2024-02-06 18:26:07 +00:00
test_minifier_common.py Enable possibly-undefined error code (#118533) 2024-01-30 21:07:01 +00:00
testing.py [dynamo] add more refleak tests (#120657) 2024-03-07 22:25:43 +00:00
trace_rules.py add int8 packed gemm support on CPU device (#118056) 2024-03-07 08:41:43 +00:00
types.py [dynamo] delete dynamo cache entry when guard function is invalidated [attempt 2] (#119107) 2024-02-07 03:32:42 +00:00
utils.py Better error messages for impl_abstract_pystub (#120959) 2024-03-04 15:24:36 +00:00