xs_courtesy c1c45e9d8e fix include path		2024-03-13 11:21:06 +08:00
..
arm	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
common	optimize rmsnorm: add vectorized elementwise op, feat loop unrolling (#5441)	2024-03-12 17:48:02 +08:00
cuda	fix include path	2024-03-13 11:21:06 +08:00
x86	refactor code	2024-03-11 17:06:57 +08:00
__init__.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00
scaled_softmax.py	[feat] refactored extension module (#5298)	2024-01-25 17:01:48 +08:00