init colossalai, support dtk2304

08f2920e · zhuwenwen · da3f0934 · 08f2920e · 08f2920e · 08f2920e
Commit 08f2920e authored Apr 23, 2023 by zhuwenwen
20 changed files
--- a/colossalai/kernel/jit/bias_gelu.py
+++ b/colossalai/kernel/jit/bias_gelu.py
--- a/colossalai/kernel/jit/option.py
+++ b/colossalai/kernel/jit/option.py
--- a/colossalai/logging/__init__.py
+++ b/colossalai/logging/__init__.py
--- a/colossalai/logging/logger.py
+++ b/colossalai/logging/logger.py
--- a/colossalai/nn/__init__.py
+++ b/colossalai/nn/__init__.py
--- a/colossalai/nn/_ops/__init__.py
+++ b/colossalai/nn/_ops/__init__.py
--- a/colossalai/nn/_ops/_utils.py
+++ b/colossalai/nn/_ops/_utils.py
--- a/colossalai/nn/_ops/addmm.py
+++ b/colossalai/nn/_ops/addmm.py
--- a/colossalai/nn/_ops/batch_norm.py
+++ b/colossalai/nn/_ops/batch_norm.py
--- a/colossalai/nn/_ops/element_wise.py
+++ b/colossalai/nn/_ops/element_wise.py
--- a/colossalai/nn/_ops/embedding.py
+++ b/colossalai/nn/_ops/embedding.py
--- a/colossalai/nn/_ops/embedding_bag.py
+++ b/colossalai/nn/_ops/embedding_bag.py
--- a/colossalai/nn/_ops/layernorm.py
+++ b/colossalai/nn/_ops/layernorm.py
--- a/colossalai/nn/_ops/linear.py
+++ b/colossalai/nn/_ops/linear.py
--- a/colossalai/nn/_ops/loss.py
+++ b/colossalai/nn/_ops/loss.py
--- a/colossalai/nn/_ops/view.py
+++ b/colossalai/nn/_ops/view.py
--- a/colossalai/nn/init.py
+++ b/colossalai/nn/init.py
--- a/colossalai/nn/layer/__init__.py
+++ b/colossalai/nn/layer/__init__.py
--- a/colossalai/nn/layer/base_layer.py
+++ b/colossalai/nn/layer/base_layer.py
--- a/colossalai/nn/layer/colossalai_layer/__init__.py
+++ b/colossalai/nn/layer/colossalai_layer/__init__.py