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.
ColossalAI/colossalai/pipeline
Super Daniel 393f594051
[fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710)
2 years ago
..
rpc [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710) 2 years ago
__init__.py [pipeline] refactor the pipeline module (#1087) 3 years ago
layer_sepc.py [pipeline] refactor the pipeline module (#1087) 3 years ago
pipelinable.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
pipeline_process_group.py [pipeline/fix-bug] num_microbatches support any integrate | stable chimera | launch tool for rpc pp framework (#1684) 2 years ago
utils.py [NFC] polish colossalai/pipeline/utils.py code style (#1562) 2 years ago