You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Frank Lee 40d376c566 [setup] support pre-build and jit-build of cuda kernels (#2374) ... * [setup] support pre-build and jit-build of cuda kernels * polish code * polish code * polish code * polish code * polish code * polish code		2 years ago
..
csrc	[hotfix] fix error for torch 2.0 (#2243)	2 years ago
__init__.py	…
flash_attention.py	updated attention kernel (#2133)	2 years ago
layer_norm.py	…
multihead_attention.py	[setup] support pre-build and jit-build of cuda kernels (#2374)	2 years ago
scaled_softmax.py	[builder] builder for scaled_upper_triang_masked_softmax (#2234)	2 years ago