Set csgmv as default lora backend. (#11488)
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user