fix: link torch_python for pybind11 symbols

2026-08-11 09:45:53 +00:00
parent 7e8605248a
commit 1cf4a34d17
1 changed files with 1 additions and 1 deletions
--- a/ex_engine/build_moe_topk.sh
+++ b/ex_engine/build_moe_topk.sh
@@ -27,7 +27,7 @@ $CXX -shared -fPIC -O2 -std=c++17 \
    -I"$TORCH_INC" \
    -I"$TORCH_INC2" \
    -L"$TORCH_LIB" \
-    -ltorch -ltorch_cpu -ltorch_cuda -lc10 -lc10_cuda \
+    -ltorch -ltorch_cpu -ltorch_cuda -ltorch_python -lc10 -lc10_cuda \
    -Wl,--no-as-needed,-rpath,"$TORCH_LIB" \
    -D_GLIBCXX_USE_CXX11_ABI=0 \
    -DTORCH_EXTENSION_NAME=moe_topk_softmax_v3 \