アマデウス 077a66dd81 updated attention kernel (#2133)

2022-12-16 10:54:03 +08:00

..

test_checkpoint

…

test_checkpoint_io

[CheckpointIO] a uniform checkpoint I/O module (#1689)

2022-11-08 15:15:13 +08:00

test_activation_checkpointing.py

[utils] Add use_reetrant=False in utils.activation_checkpoint (#1460)

2022-08-16 15:39:20 +08:00

test_colo_checkpoint.py

…

test_commons.py

…

test_flash_attention.py

updated attention kernel (#2133)

2022-12-16 10:54:03 +08:00

test_lazy_init_ctx.py

…

test_memory.py

…

test_norm_gradient_clipping.py

[Doc] add more doc for ColoTensor. (#1458)

2022-08-16 10:38:41 +08:00

test_zero_gradient_clippling.py

…