Merge branch 'main' of https://github.com/hpcaitech/ColossalAI

9e768b59 · zhuwenwen · 7bc5a8e3 · 8aed02b9 · 9e768b59 · 9e768b59
Commit 9e768b59 authored Oct 10, 2023 by zhuwenwen
20 changed files
--- a/colossalai/tensor/d_tensor/__init__.py
+++ b/colossalai/tensor/d_tensor/__init__.py
--- a/colossalai/tensor/d_tensor/api.py
+++ b/colossalai/tensor/d_tensor/api.py
--- a/colossalai/tensor/d_tensor/comm_spec.py
+++ b/colossalai/tensor/d_tensor/comm_spec.py
--- a/colossalai/tensor/d_tensor/d_tensor.py
+++ b/colossalai/tensor/d_tensor/d_tensor.py
--- a/colossalai/tensor/d_tensor/layout.py
+++ b/colossalai/tensor/d_tensor/layout.py
--- a/colossalai/tensor/d_tensor/layout_converter.py
+++ b/colossalai/tensor/d_tensor/layout_converter.py
--- a/colossalai/tensor/d_tensor/sharding_spec.py
+++ b/colossalai/tensor/d_tensor/sharding_spec.py
--- a/colossalai/tensor/d_tensor/utils.py
+++ b/colossalai/tensor/d_tensor/utils.py
--- a/colossalai/tensor/param_op_hook.py
+++ b/colossalai/tensor/param_op_hook.py
--- a/colossalai/tensor/shape_consistency.py
+++ b/colossalai/tensor/shape_consistency.py
--- a/colossalai/tensor/sharding_spec.py
+++ b/colossalai/tensor/sharding_spec.py
--- a/colossalai/tensor/utils.py
+++ b/colossalai/tensor/utils.py
--- a/colossalai/testing/__init__.py
+++ b/colossalai/testing/__init__.py
--- a/colossalai/testing/comparison.py
+++ b/colossalai/testing/comparison.py
--- a/colossalai/testing/pytest_wrapper.py
+++ b/colossalai/testing/pytest_wrapper.py
--- a/colossalai/testing/random.py
+++ b/colossalai/testing/random.py
--- a/colossalai/testing/utils.py
+++ b/colossalai/testing/utils.py
--- a/colossalai/utils/__init__.py
+++ b/colossalai/utils/__init__.py
--- a/colossalai/utils/checkpoint/__init__.py
+++ b/colossalai/utils/checkpoint/__init__.py
--- a/colossalai/utils/checkpoint_io/__init__.py
+++ b/colossalai/utils/checkpoint_io/__init__.py