github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_pipeline
History
Jianghai e7cc62d735 [pipeline] All bert models (#4233) 
* bloom policy

* llama pipeline forward and tests

* fix the output and attention_mask

* fix name

* bind argument to policy

* Revert "bloom policy"

This reverts commit 8dee68a0a2.

This policy should be revert and copied to feature/bloom

* revert the bloom changes

* cancel unneeded inputs

* gpt

* finish llama

* causal lm and sequence classification

* revision

* add pure pipeline test

* finish some bert models

* finish all bert models

* finish bert tests

* fix bugs

* fix bugs

* fix test pipeline

* fix data gen for qa

* update the set pipeline forward

* shared params

* fix bugs
 		2023-08-15 23:25:14 +08:00
..
test_policy [pipeline] All bert models (#4233) 2023-08-15 23:25:14 +08:00
test_schedule [pipeline] refactor 1f1b schedule (#4115) 2023-08-15 23:25:14 +08:00
rpc_test_utils.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
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) 2022-09-29 10:58:58 +08:00
test_cuda_rpc_optimizer.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_performance.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_pipeline.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_value_correctness.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_middleware_1f1b.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_p2p_communication.py [pipeline] implement p2p communication (#4100) 2023-08-15 23:25:14 +08:00
test_pipelinable.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_pipeline_process_group.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_stage_manager.py [pipeline]add pipeline policy and bert forward (#4130) 2023-08-15 23:25:14 +08:00