EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix test_create_kvindices unit test (#4452)

Browse Source
This commit is contained in:
lukec
2025-03-16 07:01:04 +08:00
committed by GitHub
parent 035ac2ab74
commit 21d485f835
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/srt/run_suite.py
View File

@@ -72,6 +72,7 @@ suites = {
TestFile("test_vision_openai_server.py", 344), TestFile("test_vision_openai_server.py", 344),
TestFile("test_w8a8_quantization.py", 46), TestFile("test_w8a8_quantization.py", 46),
TestFile("test_eval_fp8_accuracy.py", 172), TestFile("test_eval_fp8_accuracy.py", 172),
TestFile("test_create_kvindices.py", 2),
], ],
"nightly": [ "nightly": [
TestFile("test_nightly_gsm8k_eval.py"), TestFile("test_nightly_gsm8k_eval.py"),

4
test/srt/test_create_kvindices.py
View File

@@ -4,9 +4,7 @@ import unittest
import numpy as np import numpy as np
import torch import torch
from sglang.srt.layers.attention.flashinfer_utils import ( from sglang.srt.layers.attention.utils import create_flashinfer_kv_indices_triton
create_flashinfer_kv_indices_triton,
)
class TestCreateKvIndices(unittest.TestCase): class TestCreateKvIndices(unittest.TestCase):