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/shardformer/shard
Wang Binluo eea37da6fa
[fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016)
3 months ago
..
__init__.py
grad_ckpt_config.py
shard_config.py [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
sharder.py
shardformer.py
utils.py