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_pipeline
Ziyue Jiang 09d69e1c25
[PP Middleware] Add bwd and step for PP middleware (#2111) 		2 years ago
..
rpc_test_utils.py [PP Middleware] Add bwd and step for PP middleware (#2111) 2 years ago
test_cuda_rpc_chimera.py [pipeline/pytree] add pytree to process args and kwargs | provide `data_process_func` to process args and kwargs after forward (#1642) 2 years ago
test_cuda_rpc_optimizer.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_performance.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_pipeline.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_value_correctness.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_middleware_1f1b.py [PP Middleware] Add bwd and step for PP middleware (#2111) 2 years ago
test_pipelinable.py [pipeline] refactor the pipeline module (#1087) 2 years ago
test_pipeline_process_group.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago