4 commits

Author	SHA1	Message	Date
PanQiWei	2b532f9453	add trainable mode	2023-05-26 13:11:30 +08:00
PanQiWei	e1c564ac0e	compatible with older pytorch version	2023-05-14 16:17:03 +08:00
PanQiWei	2273f9ef39	refactor file structure for triton kernels	2023-05-14 11:49:10 +08:00
PanQiWei	fef1a4fe4b	make code clean and extendable	2023-05-12 20:11:55 +08:00