ColossalAI/colossalai/elixir/kernels
Frank Lee c173a69b3e
[elixir] refactored the chunk module (#3956)
2023-06-12 15:02:22 +08:00
..
__init__.py [elixir] refactored the chunk module (#3956) 2023-06-12 15:02:22 +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