From 1cf4a34d178dcffd33aec1d4e31017c8cdc3e630 Mon Sep 17 00:00:00 2001
From: claude <claude@project6.dev>
Date: Tue, 11 Aug 2026 09:45:53 +0000
Subject: [PATCH] fix: link torch_python for pybind11 symbols

---
 ex_engine/build_moe_topk.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ex_engine/build_moe_topk.sh b/ex_engine/build_moe_topk.sh
index 5d41da85..0d087553 100755
--- 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 \