HELSON df4f020ee3 [zero1&2] only append parameters with gradients (#2681)

2023-02-13 18:00:16 +08:00

..

_C

…

amp

[test] fixed the triton version for testing (#2608)

2023-02-07 13:49:38 +08:00

auto_parallel

[autoparallel] Patch meta information of torch.nn.functional.softmax and torch.nn.Softmax (#2674)

2023-02-13 16:09:22 +08:00

autochunk

[autochunk] support diffusion for autochunk (#2621)

2023-02-07 16:32:45 +08:00

builder

…

cli

…

communication

…

context

…

device

…

engine

…

fx

[autoparallel] Patch meta information of torch.matmul (#2584)

2023-02-08 11:05:31 +08:00

gemini

[gemini] add fake_release_chunk for keep-gathered chunk in the inference mode (#2671)

2023-02-13 14:35:32 +08:00

kernel

…

logging

…

nn

[gemini] add fake_release_chunk for keep-gathered chunk in the inference mode (#2671)

2023-02-13 14:35:32 +08:00

pipeline

…

registry

…

tensor

…

testing

…

trainer

…

utils

[gemini] fix colo_init_context (#2683)

2023-02-13 17:53:15 +08:00

zero

[zero1&2] only append parameters with gradients (#2681)

2023-02-13 18:00:16 +08:00

__init__.py

…

constants.py

…

core.py

…

global_variables.py

…

initialize.py

…