This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-05-05 12:24:38 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
25c57b9fb44e3499cb9e82bb461c3aa5a2d81a2a
ColossalAI
/
tests
/
test_utils
History
flybird1111
25c57b9fb4
[fix] coloattention support flash attention 2 (
#4347
)
...
Improved ColoAttention interface to support flash attention 2. Solved
#4322
2023-08-04 13:46:22 +08:00
..
test_checkpoint
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
test_activation_checkpointing.py
[CI] fix some spelling errors (
#3707
)
2023-05-10 17:12:03 +08:00
test_colo_checkpoint.py
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
test_commons.py
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
test_flash_attention.py
[fix] coloattention support flash attention 2 (
#4347
)
2023-08-04 13:46:22 +08:00
test_memory.py
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
test_norm_gradient_clipping.py
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
test_zero_gradient_clippling.py
[test] refactor tests with spawn (
#3452
)
2023-04-06 14:51:35 +08:00
