[sgl-kernel Code Clean] Remove useless lightning_attention kernel (#13819)
This commit is contained in:
@@ -3,12 +3,6 @@ from typing import Optional, Tuple
|
||||
import torch
|
||||
|
||||
|
||||
def lightning_attention_decode(q, k, v, past_kv, slope, output, new_kv):
|
||||
torch.ops.sgl_kernel.lightning_attention_decode.default(
|
||||
q, k, v, past_kv, slope, output, new_kv
|
||||
)
|
||||
|
||||
|
||||
def merge_state(
|
||||
v_a: torch.Tensor,
|
||||
s_a: torch.Tensor,
|
||||
|
||||
Reference in New Issue
Block a user