mirror of https://github.com/hpcaitech/ColossalAI
4b977541a8
* added softmax kernel * added qkv_kernel * added ops * adding tests * upload tets * fix tests * debugging * debugging tests * debugging * added * fixed errors * added softmax kernel * clean codes * added tests * update tests * update tests * added attention * add * fixed pytest checking * add cuda check * fix cuda version * fix typo |
||
---|---|---|
.. | ||
ops.py | ||
qkv_matmul_kernel.py | ||
softmax_kernel.py |