This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-04-25 09:12:10 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
d3c6cd66f338e5c866b997d4add9cf9b1a8be351
ColossalAI
/
tests
/
test_shardformer
History
Baizhou Zhang
da3cef27ad
[pipeline] fix return_dict/fix pure_pipeline_test (
#4331
)
2023-08-15 23:25:14 +08:00
..
test_layer
[shardformer] support inplace sharding (
#4251
)
2023-08-15 23:25:14 +08:00
test_model
[pipeline] fix return_dict/fix pure_pipeline_test (
#4331
)
2023-08-15 23:25:14 +08:00
__init__.py
[shardformer] adapted T5 and LLaMa test to use kit (
#4049
)
2023-07-04 16:05:01 +08:00
test_shard_utils.py
[test] add shard util tests
2023-08-15 23:25:14 +08:00
test_with_torch_ddp.py
[shardformer] support lazy init (
#4202
)
2023-08-15 23:25:14 +08:00