__init__.py
0 Bytes
-
shenggan authored
* add triton kernel * add softmax kernel * fix bugs in triton kernel * remove eviction_policy * refactoring triton kernel for softmax
3b046699
* add triton kernel * add softmax kernel * fix bugs in triton kernel * remove eviction_policy * refactoring triton kernel for softmax