From a2a261963f7698a730ac94f2ebd8b41cc44123c8 Mon Sep 17 00:00:00 2001 From: "Po-Han Huang (NVIDIA)" <53919306+nvpohanh@users.noreply.github.com> Date: Tue, 15 Sep 2026 21:01:05 +0800 Subject: [PATCH] [CI] Fix SWA decode radix cache NIXL import (#39600) --- .../test_disaggregation_decode_radix_cache_swa.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/test/registered/disaggregation/test_disaggregation_decode_radix_cache_swa.py b/test/registered/disaggregation/test_disaggregation_decode_radix_cache_swa.py index bc38cac44..58ab20604 100644 --- a/test/registered/disaggregation/test_disaggregation_decode_radix_cache_swa.py +++ b/test/registered/disaggregation/test_disaggregation_decode_radix_cache_swa.py @@ -9,7 +9,6 @@ import unittest from test_disaggregation_decode_radix_cache import ( DisaggregationDecodeRadixCacheTestMixin, - _has_nixl, ) 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"] +def _has_nixl(): + try: + import nixl._api # noqa: F401 + except ImportError: + return False + return True + + @unittest.skipUnless( is_in_ci() or _has_nixl(), "NIXL is required for decode radix cache disaggregation coverage.",