github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/elixir/kernels
History
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
attention.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
attn_wrapper.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
gpt_attention.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
layernorm.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00
opt_attention.py [elixir] add elixir and its unit tests (#3835) 2023-05-29 09:32:37 +08:00