github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Making large AI models cheaper, faster and more accessible
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1884 Commits
31 Branches
46 Tags
236 MiB
 
 
 
 
 
Tree: 4af31d263d
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '4af31d263d'
${ noResults }
ColossalAI/tests/test_utils
アマデウス 077a66dd81
updated attention kernel (#2133) 		2 years ago
..
test_checkpoint [hotfix] skipped unsafe test cases (#1282) 2 years ago
test_checkpoint_io [CheckpointIO] a uniform checkpoint I/O module (#1689) 2 years ago
test_activation_checkpointing.py [utils] Add use_reetrant=False in utils.activation_checkpoint (#1460) 2 years ago
test_colo_checkpoint.py [hotfix] fix a running error in test_colo_checkpoint.py (#1387) 2 years ago
test_commons.py [gemini] add GeminiMemoryManger (#832) 3 years ago
test_flash_attention.py updated attention kernel (#2133) 2 years ago
test_lazy_init_ctx.py [utils] integrated colotensor with lazy init context (#1324) 2 years ago
test_memory.py [test] ignore 8 gpu test (#1080) 2 years ago
test_norm_gradient_clipping.py [Doc] add more doc for ColoTensor. (#1458) 2 years ago
test_zero_gradient_clippling.py [test] refactored with the new rerun decorator (#763) 3 years ago