Skip to content

GitLab

Unverified Commit 21d485f8 authored by lukec's avatar lukec Committed by GitHub
Browse files

Fix test_create_kvindices unit test (#4452)

parent 035ac2ab
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
test/srt/run_suite.py
View file @ 21d485f8
...@@ -72,6 +72,7 @@ suites = { ...@@ -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"),
......
test/srt/test_create_kvindices.py
View file @ 21d485f8
...@@ -4,9 +4,7 @@ import unittest ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment