dylanyunlong/project_6
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix: import torch.utils.cpp_extension explicitly in build script

Browse Source
This commit is contained in:
project6-dev
2026-08-13 11:37:10 +00:00
parent f6cf9d662e
commit 6092edebde
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qwen3_6_scripts/build_cccl_moe_sort_scatter.sh
View File

@@ -20,8 +20,8 @@ done
[[ -n "${CXX}" ]] || { echo "no corex clang++"; exit 2; }
# Find torch paths
TORCH_INC=$(python3 -c "import torch; print(torch.utils.cpp_extension.include_paths()[0])")
TORCH_LIB=$(python3 -c "import torch.utils.cpp_extension as e; import os; print(os.path.join(os.path.dirname(e.__file__), '..', '..', 'lib'))" | xargs realpath)
TORCH_INC=$(python3 -c "from torch.utils.cpp_extension import include_paths; print(include_paths()[0])")
TORCH_LIB=$(python3 -c "import torch; import os; print(os.path.join(os.path.dirname(torch.__file__), 'lib'))")
PYTHON_INC=$(python3 -c "from sysconfig import get_paths; print(get_paths()['include'])")
CUDA_INC="/usr/local/corex/include"