6 commits

Author	SHA1	Message	Date
qwopqwop200	b3654a68c3	support 32dim triton kernel	2023-06-02 19:04:12 +09:00
PanQiWei	2b532f9453	add trainable mode	2023-05-26 13:11:30 +08:00
qwopqwop200	503f85255d	Update kernels.py	2023-05-25 23:15:33 +09:00
PanQiWei	86b3b52c63	fix ImportError when triton is not installed	2023-05-20 16:15:20 +08:00
PanQiWei	de33d26d67	fix bugs	2023-05-14 13:07:18 +08:00
PanQiWei	2273f9ef39	refactor file structure for triton kernels	2023-05-14 11:49:10 +08:00