perf(mamba): use Triton conv1d for non-contiguous input to avoid .contiguous() copy (#20469)

This commit is contained in:
Jia Guo
2026-03-19 19:38:46 -07:00
committed by GitHub
parent db995fba47
commit 87549f8f0b
2 changed files with 21 additions and 16 deletions
+2 -2
View File
@@ -118,7 +118,7 @@ def run_eval(args):
elif args.eval_name == "gpqa":
from sglang.test.simple_eval_gpqa import GPQAEval
filename = (
filename = getattr(args, "dataset_path", None) or (
"https://openaipublic.blob.core.windows.net/simple-evals/gpqa_diamond.csv"
)
eval_obj = GPQAEval(filename, args.num_examples, args.num_threads)
@@ -130,7 +130,7 @@ def run_eval(args):
from sglang.test.simple_eval_longbench_v2 import LongBenchV2Eval
# Default to HuggingFace dataset, can be overridden with --dataset-path
data_source = args.dataset_path
data_source = getattr(args, "dataset_path", None)
categories = args.categories.split(",") if args.categories else None
eval_obj = LongBenchV2Eval(