Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
8ce68370b5
commit
99f636a86f
@@ -4,7 +4,9 @@ import argparse
|
||||
|
||||
import torch
|
||||
|
||||
from sglang.jit_kernel.triton.gdn_fused_proj import fused_qkv_split_gdn_prefill
|
||||
from sglang.kernels.ops.attention.triton_gdn_fused_proj import (
|
||||
fused_qkv_split_gdn_prefill,
|
||||
)
|
||||
|
||||
DTYPES = {
|
||||
"bf16": torch.bfloat16,
|
||||
|
||||
Reference in New Issue
Block a user