| .. |
|
__init__.py
|
dynamo: handle DTensor.device_mesh.device_type (#118803)
|
2024-03-22 14:42:22 +00:00 |
|
base.py
|
[dynamo] Replace VariableTracker.apply with visit/realize_all (#122218)
|
2024-03-20 07:53:18 +00:00 |
|
builder.py
|
Proper view support for jagged layout NestedTensor (#113279)
|
2024-03-22 02:12:36 +00:00 |
|
builtin.py
|
dynamo: handle DTensor.device_mesh.device_type (#118803)
|
2024-03-22 14:42:22 +00:00 |
|
constant.py
|
[dynamo] Remove VariableTracker.parents_tracker (#122058)
|
2024-03-19 04:23:24 +00:00 |
|
ctx_manager.py
|
Revert "Teach dynamo about torch.func.jvp (#119926)"
|
2024-03-20 18:34:43 +00:00 |
|
dicts.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
distributed.py
|
dynamo: handle DTensor.device_mesh.device_type (#118803)
|
2024-03-22 14:42:22 +00:00 |
|
functions.py
|
Handle special kwargs in user-written Triton kernel calls (#122280)
|
2024-03-21 03:34:07 +00:00 |
|
higher_order_ops.py
|
[while_loop] disable closure capturing and manually set the inputs. (#122244)
|
2024-03-20 20:14:35 +00:00 |
|
iter.py
|
[dynamo] Optimize SourcelessBuilder (#122063)
|
2024-03-19 04:23:30 +00:00 |
|
lazy.py
|
[dynamo] Replace VariableTracker.apply with visit/realize_all (#122218)
|
2024-03-20 07:53:18 +00:00 |
|
lists.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
misc.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
nn_module.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
optimizer.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
sdpa.py
|
[dynamo] Refactor reconstruct() not to return anything (#120150)
|
2024-02-17 17:13:41 +00:00 |
|
tensor.py
|
[dynamo] Add missing _nonvar_fields annotations (#122219)
|
2024-03-20 07:53:18 +00:00 |
|
torch_function.py
|
[dynamo] Optimize SourcelessBuilder (#122063)
|
2024-03-19 04:23:30 +00:00 |
|
torch.py
|
Revert "Teach dynamo about torch.func.jvp (#119926)"
|
2024-03-20 18:34:43 +00:00 |
|
user_defined.py
|
[dynamo] Optimize SourcelessBuilder (#122063)
|
2024-03-19 04:23:30 +00:00 |
2e44b12dd4