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.
551 Commits
31 Branches
46 Tags
236 MiB
 
 
 
 
 
Tree: 676f191532
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '676f191532'
${ noResults }
ColossalAI/tests
Jiarui Fang e76f76c08b
[Tensor] test parameters() as member function (#896) 		3 years ago
..
components_to_test [tensor] add cross_entrophy_loss (#868) 3 years ago
test_amp [test] refactored with the new rerun decorator (#763) 3 years ago
test_comm [test] refactored with the new rerun decorator (#763) 3 years ago
test_config
…
test_context [test] refactored with the new rerun decorator (#763) 3 years ago
test_data [unittest] refactored unit tests for change in dependency (#838) 3 years ago
test_data_pipeline_tensor_parallel [refactor] moving grad acc logic to engine (#804) 3 years ago
test_engine [refactor] moving grad acc logic to engine (#804) 3 years ago
test_gemini [gemini] accelerate adjust_layout() (#878) 3 years ago
test_layers [test] refactored with the new rerun decorator (#763) 3 years ago
test_moe [test] refactored with the new rerun decorator (#763) 3 years ago
test_optimizer
…
test_tensor [Tensor] test parameters() as member function (#896) 3 years ago
test_trainer [test] refactored with the new rerun decorator (#763) 3 years ago
test_utils [gemini] add GeminiMemoryManger (#832) 3 years ago
test_zero [gemini] add GeminiMemoryManger (#832) 3 years ago
__init__.py
…