| .. |
|
__init__.py
|
[dynamo] implement custom dict variable as a general solution for HF's ModelOutput class (#105044)
|
2023-09-14 17:15:50 +00:00 |
|
base.py
|
Back out "Dynamo support for autograd.Function w/ once_differentiable (#108686)" (#109199)
|
2023-09-13 15:43:20 +00:00 |
|
builder.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
builtin.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
constant.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
ctx_manager.py
|
[dynamo] remove DummyGlobalSource (#109411)
|
2023-09-16 23:11:11 +00:00 |
|
dicts.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
distributed.py
|
[dynamo] fix dynamo + DTensor to work with 2d (#108329)
|
2023-08-31 22:46:26 +00:00 |
|
functions.py
|
Support register_hook on input tensors (#108903)
|
2023-09-14 01:52:21 +00:00 |
|
higher_order_ops.py
|
Trace pytree calls inside vmap implementation. (#109107)
|
2023-09-20 18:11:10 +00:00 |
|
lists.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
misc.py
|
[dynamo] implement custom dict variable as a general solution for HF's ModelOutput class (#105044)
|
2023-09-14 17:15:50 +00:00 |
|
nn_module.py
|
Functools partial support in dynamo (#108846)
|
2023-09-09 17:25:02 +00:00 |
|
optimizer.py
|
Cudagraphs support for compiled optimizers (#107504)
|
2023-08-31 20:47:18 +00:00 |
|
tensor.py
|
Fix numpy(force=False) (#109634)
|
2023-09-20 20:06:13 +00:00 |
|
torch.py
|
Trace through pytree API with dynamo. (#108533)
|
2023-09-20 00:04:56 +00:00 |
|
user_defined.py
|
[RELAND] Force synced KJT to trace unbacked SymInt (#108960) (#109216)
|
2023-09-18 14:39:44 +00:00 |
1f6828ca99