Set csgmv as default lora backend. (#11488)

This commit is contained in:
Lifu Huang
2025-10-15 21:53:24 -07:00
committed by GitHub
parent cbac499750
commit b0d20cdec7
11 changed files with 11 additions and 23 deletions

View File

@@ -14,7 +14,7 @@
import dataclasses
import random
from typing import List
from typing import List, Optional
import torch
@@ -50,7 +50,7 @@ class LoRAModelCase:
TORCH_DTYPES = [torch.float16]
BACKENDS = ["triton"]
BACKENDS = ["triton", "csgmv"]
DEFAULT_PROMPTS = [
"AI is a field of computer science focused on",
"""
@@ -135,7 +135,7 @@ def run_lora_test_one_by_one(
model_case: LoRAModelCase,
torch_dtype: torch.dtype,
max_new_tokens: int,
backend: str,
backend: str = "csgmv",
disable_cuda_graph: bool = False,
disable_radix_cache: bool = False,
mem_fraction_static: float = 0.88,