Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 16:46:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ab54fed29295c4aa2675635cf2dadc5f9281171f
ColossalAI/colossalai/nn/_ops
History
Tongping Liu ab54fed292 [hotfix] add kwargs for colo_addmm (#2171)
2022-12-22 13:25:30 +08:00
..
__init__.py
[Gemini] patch for supporting orch.add_ function for ColoTensor (#2003)
2022-11-25 20:06:35 +08:00
_utils.py
[embedding] tablewise sharding polish (#1535)
2022-09-02 11:09:37 +08:00
addmm.py
[hotfix] add kwargs for colo_addmm (#2171)
2022-12-22 13:25:30 +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_bag.py
[NFC] polish colossalai/nn/_ops/embedding_bag.py code style (#1552)
2022-09-08 22:11:04 +08:00
embedding.py
[NFC] polish colossalai/nn/_ops/embedding.py code style (#1561)
2022-09-08 22:11:04 +08:00
layernorm.py
[NFC] polish colossalai/nn/_ops/layernorm.py code style (#1555)
2022-09-08 22:11:04 +08:00
linear.py
[tensor] added linear implementation for the new sharding spec (#1416)
2022-08-12 11:33:09 +08:00
loss.py
[tensor] fix a assertion in colo_tensor cross_entropy (#1232)
2022-07-08 11:18:00 +08:00
view.py
[colotensor] add Tensor.view op and its unit test (#1343)
2022-07-21 10:53:15 +08:00
Powered by Gitea Version: 1.25.2 Page: 872ms Template: 46ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API