ColossalAI/colossalai/kernel
Frank Lee 95a36eae63
[kernel] added kernel loader to softmax autograd function (#3093)
* [kernel] added kernel loader to softmax autograd function

* [release] v0.2.6
2023-03-10 14:27:09 +08:00
..
cuda_native [kernel] added kernel loader to softmax autograd function (#3093) 2023-03-10 14:27:09 +08:00
jit [misc] add reference (#2930) 2023-02-28 18:07:24 +08:00
__init__.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
op_builder