diff --git a/qwen3_6_scripts/patch_ops.sh b/qwen3_6_scripts/patch_ops.sh index 3f70c7cf..28a794c0 100755 --- a/qwen3_6_scripts/patch_ops.sh +++ b/qwen3_6_scripts/patch_ops.sh @@ -246,6 +246,20 @@ if source != installed: raise SystemExit("runtime api_server overlay identity mismatch") PY +build_stage "compiling CCCL CachingDeviceAllocator LD_PRELOAD module" +bash ./cccl_preload/build_cccl_preload.sh /workspace/qwen3_6_scripts/cccl_preload || \ + echo "[WARN] CCCL preload allocator build failed — will use default allocator" + +build_stage "compiling CoreX CUDA extensions (moe_index_combine + gdn_chunk_recurrent)" +if [[ -x /usr/local/corex-3.2.3/bin/clang++ ]]; then + bash ./build_corex_moe_index_combine.sh "${VLLM_ROOT}" || \ + echo "[WARN] moe_index_combine build failed — will use PyTorch fallback" + bash ./build_corex_gdn_chunk_recurrent.sh "${VLLM_ROOT}" || \ + echo "[WARN] gdn_chunk_recurrent build failed — will use Python fallback" +else + echo "[WARN] corex clang++ not found — skipping extension builds" +fi + build_stage "compiling submission Python sources" find . -path './wheels' -prune -o -name '*.py' -print0 | xargs -0 python3 -m py_compile build_stage "patch script completed"