ColossalAI/colossalai/nn/_ops
digger-yu b9a8dff7e5
[doc] Fix typo under colossalai and doc(#3618)
* Fixed several spelling errors under colossalai

* Fix the spelling error in colossalai and docs directory

* Cautious Changed the spelling error under the example folder

* Update runtime_preparation_pass.py

revert autograft to autograd

* Update search_chunk.py

utile to until

* Update check_installation.py

change misteach to mismatch in line 91

* Update 1D_tensor_parallel.md

revert to perceptron

* Update 2D_tensor_parallel.md

revert to perceptron in line 73

* Update 2p5D_tensor_parallel.md

revert to perceptron in line 71

* Update 3D_tensor_parallel.md

revert to perceptron in line 80

* Update README.md

revert to resnet in line 42

* Update reorder_graph.py

revert to indice in line 7

* Update p2p.py

revert to megatron in line 94

* Update initialize.py

revert to torchrun in line 198

* Update routers.py

change to detailed in line 63

* Update routers.py

change to detailed in line 146

* Update README.md

revert  random number in line 402
2023-04-26 11:38:43 +08:00
..
__init__.py [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003) 2022-11-25 20:06:35 +08:00
_utils.py [NFC] polish code style (#3268) 2023-03-29 15:22:21 +08:00
addmm.py [NFC] polish colossalai/nn/_ops/addmm.py code style (#3274) 2023-03-29 15:22:21 +08:00
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 [doc] Fix typo under colossalai and doc(#3618) 2023-04-26 11:38:43 +08:00
layernorm.py
linear.py [zero] fix error for BEiT models (#2169) 2022-12-26 15:03:54 +08:00
loss.py
view.py [hotfix]: Remove math.prod dependency (#2837) 2023-02-23 23:56:15 +08:00