[CI][Bugfix] Update IPC test caller for #57312's _apply_entries signature (#58107)
Signed-off-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com> Co-authored-by: kimi-no-na-wa <kimi-no-na-wa@users.noreply.github.com> Co-authored-by: Kevin H. Luu <khluu000@gmail.com>
vllm-agent committed
Sep 22, 2026 at 18:50 UTC
8d16ca6cc21931481f6c6f9ec0365874a80c6bb2
1 file changed
+4
-1
tests/kernels/moe/test_flashinfer.py
+4
-1
@@ -623,6 +623,9 @@ def test_unquantized_flashinfer_trtllm_weights_can_be_reprocessed(
623
624
def _check_flashinfer_ipc_weights(entries, expected, mode, device_index):
625
from vllm.model_executor.model_loader.weight_cache.ipc_loader import IpcModelLoader
626
+ from vllm.model_executor.model_loader.weight_cache.protocol import (
627
+ WeightCacheState,
628
+ )
629
from vllm.model_executor.utils import weights_already_processed
630
631
torch.accelerator.set_device_index(device_index)
@@ -632,7 +635,7 @@ def _check_flashinfer_ipc_weights(entries, expected, mode, device_index):
635
)
636
loader = object.__new__(IpcModelLoader)
637
loader.mode = mode
635
- loader._apply_entries(layer, entries, {}, device_index)
638
+ loader._apply_entries(layer, WeightCacheState(entries, {}, {}), device_index)
639
pointers = (layer.w13_weight.data_ptr(), layer.w2_weight.data_ptr())
640
641
# Like ipc_cache: a fresh method, only tensor metadata, no _setup_kernel.