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.

Yuanheng Zhao d63c469f45 [Infer] Revise and Adapt Triton Kernels for Spec-Dec (#5401) ... * [Infer/Fix] Fix Dependency in test - RMSNorm kernel (#5399) fix dependency in pytest * resolve conflicts for revising flash-attn * adapt kv cache copy kernel for spec-dec * fix seqlen-n kvcache copy kernel/tests * test kvcache copy - use torch.equal * add assertions * (trivial) comment out		8 months ago
..
jit	[npu] change device to accelerator api (#5239)	11 months ago
triton	[Infer] Revise and Adapt Triton Kernels for Spec-Dec (#5401)	8 months ago
__init__.py	[feat] refactored extension module (#5298)	10 months ago
extensions	[feat] refactored extension module (#5298)	10 months ago
kernel_loader.py	[Fix] resolve conflicts of merging main	8 months ago