ColossalAI/colossalai/nn/_ops
Jiarui Fang 96134e7be3
[hotfix] add bert test for gemini fwd bwd (#2035)
2022-11-29 11:19:52 +08:00
..
__init__.py [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003) 2022-11-25 20:06:35 +08:00
_utils.py
addmm.py
batch_norm.py [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003) 2022-11-25 20:06:35 +08:00
element_wise.py [hotfix] add bert test for gemini fwd bwd (#2035) 2022-11-29 11:19:52 +08:00
embedding.py
embedding_bag.py
layernorm.py
linear.py
loss.py
view.py