pytorch/torch/_dynamo
2024-03-12 21:31:18 +00:00
..
backends Switch cudagraph backend to cudagraph trees (#121019) 2024-03-08 22:56:26 +00:00
repro Add missing newline to repro and some utility thing in repro (#121051) 2024-03-04 22:52:54 +00:00
variables Revert "[dynamo][guards] Use lazy variable tracker for func defaults (#121388)" 2024-03-12 21:31:18 +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 Improve Dynamo support for torch function and class methods in general (#121365) 2024-03-08 20:03:49 +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 [dynamo] preserve some FX node metadata of GraphModules (#107067) 2023-09-15 23:29:14 +00:00
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 Add Cudagraphs disable checking (#121018) 2024-03-08 22:47:24 +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 [HigherOrderOp] Fall back on all new side effects in speculate_subgraph (#104077) 2023-06-28 14:20:37 +00:00
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 feat: Add min, max ranges to mark_dynamic API (#119737) 2024-03-07 23:26:03 +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 make dyanmo work with _LazyGraphModule.lazy_forward (#121259) 2024-03-08 01:37:39 +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 [dynamo] Enable typechecking for funcname_cache.py (#112031) 2023-10-26 04:54:16 +00:00
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] Tweak naming for module hook bw_state (#121609) 2024-03-12 16:27:56 +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 [dynamo] Fix support for nn.Parameter constructor (part 1) (#120163) 2024-03-11 05:14:42 +00:00
symbolic_convert.py Revert "Limit loop unrolling (#120023)" 2024-03-02 00:04:08 +00:00
tensor_version_op.py [dynamo] Support _unsafe_set_version_counter (#121086) 2024-03-11 05:14:42 +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 Revert "Batch Norm Consolidation (#116092)" 2024-03-11 22:22:41 +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 [eazy] add more log when accuracy check fail (#121656) 2024-03-12 20:58:20 +00:00