Fix import path for flashinfer_api

djns99 · djns99 · commit f93dd1436b42 · 2025-12-10T15:34:14.000+13:00
diff --git a/flashinfer/comm/trtllm_moe_alltoall.py b/flashinfer/comm/trtllm_moe_alltoall.py
@@ -12,7 +12,7 @@
 import torch
 import functools
 
-from .api_logging import flashinfer_api
+from ..api_logging import flashinfer_api
 
 from .mnnvl import MnnvlMemory, MnnvlConfig
 from .mapping import Mapping
