ColossalAI/tests/test_tensor/test_dtensor at 2ef855c79885c028dccf40172125f7a63e30e4a0 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 09:24:09 +00:00

Files

History

YuliangLiu0306 29386a54e6 [DTensor] refactor CommSpec (#3034 )

2023-03-08 10:45:31 +08:00

..

test_comm_spec.py

[DTensor] refactor CommSpec (#3034 )

2023-03-08 10:45:31 +08:00

test_dtensor_sharding_spec.py

[hotfix] skip auto checkpointing tests (#3029 )

2023-03-07 15:50:00 +08:00

test_dtensor.py

[DTensor] refactor sharding spec (#2987 )

2023-03-07 11:08:11 +08:00