Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-11-28 12:19:27 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
018936a3f30e8a08e7f99e094340ce322b0c2966
ColossalAI/tests/test_tensor
History
YuliangLiu0306 8e4e8601b7 [DTensor] implement layout converter (#3055)
* [DTensor] refactor LayoutConverter for DTensor

* polish code

* polish docstring
2023-03-10 09:53:52 +08:00
..
common_utils
[zero] add unit testings for hybrid parallelism (#2486)
2023-01-18 10:36:10 +08:00
core
…
model
[Gemini] remove static tracer (#2083)
2022-12-06 12:53:58 +08:00
test_dtensor
[DTensor] implement layout converter (#3055)
2023-03-10 09:53:52 +08:00
test_colo_checkpoint_tools.py
…
test_comm_spec_apply.py
[autoparallel] shard param and buffer as expected (#1753)
2022-10-21 15:45:13 +08:00
test_context.py
[ColoTensor] reconfig ColoInitContext, decouple default_pg and default_dist_spec. (#1953)
2022-11-15 16:24:16 +08:00
test_mix_gather.py
[autoparallel] mix gather (#1977)
2022-11-23 21:49:17 +08:00
test_parameter.py
…
test_shape_consistency_apply.py
[autoparallel] shard param and buffer as expected (#1753)
2022-10-21 15:45:13 +08:00
test_shape_consistency.py
[autoparallel] update CommSpec (#1667)
2022-09-29 11:20:59 +08:00
test_sharded_linear.py
[ColoTensor] ColoInitContext initialize parameters in shard mode. (#1937)
2022-11-14 16:05:09 +08:00
test_sharding_spec.py
[autoparallel] handled illegal sharding strategy (#1728)
2022-10-19 12:53:06 +08:00
test_tp_with_zero.py
[gemini] update ddp strict mode (#2518)
2023-01-28 14:35:25 +08:00
Powered by Gitea Version: 1.24.5 Page: 1839ms Template: 195ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API