pytorch/torch/_dynamo/variables
2024-08-19 17:21:48 +00:00
..
__init__.py [Dynamo] Support torch.cuda.device ctx manager (#133385) 2024-08-16 17:05:55 +00:00
base.py [dynamo] revert map/zip iterator related changes (#132528) 2024-08-04 18:46:55 +00:00
builder.py [Traceable FSDP2][Dynamo] Support reconstructing CUDA event object within Dynamo graph (#133635) 2024-08-16 20:40:46 +00:00
builtin.py [refactor][dynamo][side-effects] Helper function for __new__ for user defined class (#133799) 2024-08-19 17:21:48 +00:00
constant.py [easy][dynamo] Add tx as an arg in getitem_const (#132899) 2024-08-07 21:35:41 +00:00
ctx_manager.py [Dynamo] Support torch.cuda.device ctx manager (#133385) 2024-08-16 17:05:55 +00:00
dicts.py [BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200) 2024-08-15 15:50:19 +00:00
distributed.py Revert "[dtensor] move DTensor to public namespace (#133113)" 2024-08-19 05:00:19 +00:00
functions.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
higher_order_ops.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
iter.py [dynamo] revert map/zip iterator related changes (#132528) 2024-08-04 18:46:55 +00:00
lazy.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
lists.py [dynamo] Support object creation of classes with custom __new__ (#132977) 2024-08-16 03:09:23 +00:00
misc.py [refactor][dynamo][side-effects] Helper function for __new__ for user defined class (#133799) 2024-08-19 17:21:48 +00:00
nn_module.py Revert #132806, #132736, #132539, #132487 (#133570) 2024-08-15 20:54:21 +00:00
optimizer.py [Inductor UT] Reuse inductor UT for intel GPU test/inductor/test_compiled_optimizers.py (#133083) 2024-08-17 01:15:26 +00:00
script_object.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
sdpa.py Grouped Query Attention (#132689) 2024-08-07 05:35:36 +00:00
tensor.py Add basic mypy annotations to dynamo (#132415) 2024-08-04 18:43:36 +00:00
torch_function.py Fix _dynamo.variables.torch_function.global_mangled_class_name (#132744) 2024-08-09 22:19:01 +00:00
torch.py Revert "[dtensor] move DTensor to public namespace (#133113)" 2024-08-19 05:00:19 +00:00
user_defined.py [refactor][dynamo][side-effects] Helper function for __new__ for user defined class (#133799) 2024-08-19 17:21:48 +00:00