support dtk2210

0b58bdea · huchen · c4dd1fd4 · 0b58bdea · 0b58bdea · 0b58bdea
Commit 0b58bdea authored Nov 17, 2022 by huchen
20 changed files
--- a/colossalai/utils/__pycache__/__init__.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/__init__.cpython-37.pyc
--- a/colossalai/utils/__pycache__/activation_checkpoint.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/activation_checkpoint.cpython-37.pyc
--- a/colossalai/utils/__pycache__/checkpointing.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/checkpointing.cpython-37.pyc
--- a/colossalai/utils/__pycache__/common.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/common.cpython-37.pyc
--- a/colossalai/utils/__pycache__/cuda.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/cuda.cpython-37.pyc
--- a/colossalai/utils/__pycache__/memory.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/memory.cpython-37.pyc
--- a/colossalai/utils/__pycache__/timer.cpython-37.pyc
+++ b/colossalai/utils/__pycache__/timer.cpython-37.pyc
--- a/colossalai/utils/data_sampler/__pycache__/__init__.cpython-37.pyc
+++ b/colossalai/utils/data_sampler/__pycache__/__init__.cpython-37.pyc
--- a/colossalai/utils/data_sampler/__pycache__/base_sampler.cpython-37.pyc
+++ b/colossalai/utils/data_sampler/__pycache__/base_sampler.cpython-37.pyc
--- a/colossalai/utils/data_sampler/__pycache__/data_parallel_sampler.cpython-37.pyc
+++ b/colossalai/utils/data_sampler/__pycache__/data_parallel_sampler.cpython-37.pyc
--- a/colossalai/utils/gradient_accumulation/__pycache__/__init__.cpython-37.pyc
+++ b/colossalai/utils/gradient_accumulation/__pycache__/__init__.cpython-37.pyc
--- a/colossalai/utils/gradient_accumulation/__pycache__/_gradient_accumulation.cpython-37.pyc
+++ b/colossalai/utils/gradient_accumulation/__pycache__/_gradient_accumulation.cpython-37.pyc
--- a/colossalai/utils/multi_tensor_apply/__pycache__/__init__.cpython-37.pyc
+++ b/colossalai/utils/multi_tensor_apply/__pycache__/__init__.cpython-37.pyc
--- a/colossalai/utils/multi_tensor_apply/__pycache__/multi_tensor_apply.cpython-37.pyc
+++ b/colossalai/utils/multi_tensor_apply/__pycache__/multi_tensor_apply.cpython-37.pyc
--- a/colossalai/zero/__pycache__/__init__.cpython-37.pyc
+++ b/colossalai/zero/__pycache__/__init__.cpython-37.pyc
--- a/colossalai/zero/__pycache__/loss_scaler.cpython-37.pyc
+++ b/colossalai/zero/__pycache__/loss_scaler.cpython-37.pyc
--- a/colossalai/zero/__pycache__/zero_redundancy_optimizer_level_2.cpython-37.pyc
+++ b/colossalai/zero/__pycache__/zero_redundancy_optimizer_level_2.cpython-37.pyc
--- a/colossalai/zero/__pycache__/zero_redundancy_optimizer_level_3.cpython-37.pyc
+++ b/colossalai/zero/__pycache__/zero_redundancy_optimizer_level_3.cpython-37.pyc
--- a/compile.sh
+++ b/compile.sh
--- a/dist/colossalai-0.0.2-cp37-cp37m-linux_x86_64.whl
+++ b/dist/colossalai-0.0.2-cp37-cp37m-linux_x86_64.whl