[CI] Fix SWA decode radix cache NIXL import (#39600)
This commit is contained in:
@@ -9,7 +9,6 @@ import unittest
|
|||||||
|
|
||||||
from test_disaggregation_decode_radix_cache import (
|
from test_disaggregation_decode_radix_cache import (
|
||||||
DisaggregationDecodeRadixCacheTestMixin,
|
DisaggregationDecodeRadixCacheTestMixin,
|
||||||
_has_nixl,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
from sglang.test.ci.ci_register import register_cuda_ci
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
@@ -23,6 +22,14 @@ register_cuda_ci(est_time=365, stage="extra-b", runner_config="8-gpu-h200")
|
|||||||
SWA_SERVER_ARGS = ["--page-size", "64", "--attention-backend", "triton"]
|
SWA_SERVER_ARGS = ["--page-size", "64", "--attention-backend", "triton"]
|
||||||
|
|
||||||
|
|
||||||
|
def _has_nixl():
|
||||||
|
try:
|
||||||
|
import nixl._api # noqa: F401
|
||||||
|
except ImportError:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
@unittest.skipUnless(
|
@unittest.skipUnless(
|
||||||
is_in_ci() or _has_nixl(),
|
is_in_ci() or _has_nixl(),
|
||||||
"NIXL is required for decode radix cache disaggregation coverage.",
|
"NIXL is required for decode radix cache disaggregation coverage.",
|
||||||
|
|||||||
Reference in New Issue
Block a user