githubmirror/ColossalAI | Gitverse

/

/

Обзор ДокументацияВойти

/

/

Аналитика

Безопасность

tests/test_pipeline/

rpc_test_utils.py

[PP Middleware] Add bwd and step for PP middleware (#2111)

4 года назад

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)

4 года назад

test_cuda_rpc_optimizer.py

[pipeline/chimera] test chimera | fix bug of initializing (#1615)

4 года назад

test_cuda_rpc_performance.py

[pipeline/chimera] test chimera | fix bug of initializing (#1615)

4 года назад

test_cuda_rpc_pipeline.py

[pipeline/chimera] test chimera | fix bug of initializing (#1615)

4 года назад

test_cuda_rpc_value_correctness.py

[pipeline/chimera] test chimera | fix bug of initializing (#1615)

4 года назад

test_middleware_1f1b.py

[PP Middleware] Add bwd and step for PP middleware (#2111)

4 года назад

test_pipelinable.py

[pipeline] refactor the pipeline module (#1087)

4 года назад

test_pipeline_process_group.py

[pipeline/chimera] test chimera | fix bug of initializing (#1615)

4 года назад