ColossalAI/colossalai/kernel/cuda_native
ver217 823f3b9cf4
[doc] add deepspeed citation and copyright (#2996)
* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright
2023-03-04 20:08:11 +08:00
..
csrc [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
__init__.py [kernel] fixed repeated loading of kernels (#2549) 2023-02-03 09:47:13 +08:00
flash_attention.py [triton] added copyright information for flash attention (#2835) 2023-02-21 11:25:57 +08:00
layer_norm.py [kernel] fixed repeated loading of kernels (#2549) 2023-02-03 09:47:13 +08:00
multihead_attention.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
scaled_softmax.py [misc] add reference (#2930) 2023-02-28 18:07:24 +08:00