| .. |
|
__init__.py
|
[dynamo] support torch.nn.attention.sdpa_kernel context manager (#135404)
|
2024-09-12 22:04:48 +00:00 |
|
base.py
|
[dynamo] reland map/zip iterator related changes (#135074)
|
2024-09-06 20:38:02 +00:00 |
|
builder.py
|
Optimize dict reconstruct to not codegen untouched values (#134876)
|
2024-09-23 21:45:44 +00:00 |
|
builtin.py
|
[dynamo] simplify implementation for builtins.sum (#133779)
|
2024-09-16 04:53:06 +00:00 |
|
constant.py
|
[dynamo] Fix support for classmethod(property(...)) (#134968)
|
2024-09-18 04:47:51 +00:00 |
|
ctx_manager.py
|
[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)
|
2024-09-14 18:52:22 +00:00 |
|
dicts.py
|
Remove vt argument in raise_observed_exception (#136037)
|
2024-09-24 02:36:57 +00:00 |
|
distributed.py
|
[reland][dtensor] move DTensor to public namespace (#134203)
|
2024-09-08 17:08:40 +00:00 |
|
functions.py
|
[Traceable FSDP2] Ignore FSDP2 forward hook side-effects in AC; Support FSDP2 + AC (#134997)
|
2024-09-15 02:00:17 +00:00 |
|
higher_order_ops.py
|
[Traceable FSDP2] Ignore FSDP2 forward hook side-effects in AC; Support FSDP2 + AC (#134997)
|
2024-09-15 02:00:17 +00:00 |
|
iter.py
|
Remove vt argument in raise_observed_exception (#136037)
|
2024-09-24 02:36:57 +00:00 |
|
lazy.py
|
Type _dynamo/variables/lazy.py (#136376)
|
2024-09-21 22:18:02 +00:00 |
|
lists.py
|
Remove vt argument in raise_observed_exception (#136037)
|
2024-09-24 02:36:57 +00:00 |
|
misc.py
|
Implement GetAttrVariable.as_python_constant() (#134216)
|
2024-09-20 03:44:43 +00:00 |
|
nn_module.py
|
Remove vt argument in raise_observed_exception (#136037)
|
2024-09-24 02:36:57 +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
|
[Traceable FSDP2][Partitioner] Must save AC output if output has a backward hook (#135727)
|
2024-09-14 08:45:58 +00:00 |
|
torch_function.py
|
[Dynamo] Remove ignored modes workaround (#135502)
|
2024-09-14 18:52:22 +00:00 |
|
torch.py
|
[Dynamo] Trace enter/exit of TorchFunctionModes (#135422)
|
2024-09-14 18:52:22 +00:00 |
|
user_defined.py
|
Remove vt argument in raise_observed_exception (#136037)
|
2024-09-24 02:36:57 +00:00 |
e09c5b6046