ColossalAI/tests/test_elixir/utils
Haichen Huang 206280408a
[elixir] add elixir and its unit tests (#3835)
* [elixir] add elixir

* [elixir] add unit tests

* remove useless code

* fix python 3.8 issue

* fix typo

* add test skip

* add docstrings

* add docstrings

* add readme

* fix typo
2023-05-29 09:32:37 +08:00
..
__init__.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
gpt.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
mlp.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
opt.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
registry.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
resnet.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
small.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00