From 061a7805e4a87cfb5238260692e317c207a4772a Mon Sep 17 00:00:00 2001
From: zhyncs <me@zhyncs.com>
Date: Wed, 15 Jan 2025 20:37:32 +0000
Subject: [PATCH] upd

---
 sgl-kernel/build.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh
index 7820e401ed2..55ce9df7f33 100755
--- a/sgl-kernel/build.sh
+++ b/sgl-kernel/build.sh
@@ -11,6 +11,7 @@ docker run --rm \
     ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.} && \
     export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
     export CUDA_VERSION=${CUDA_VERSION} && \
+    mkdir -p /usr/lib/x86_64-linux-gnu/ && \
     ln -s /usr/local/cuda-${CUDA_VERSION}/targets/x86_64-linux/lib/stubs/libcuda.so /usr/lib/x86_64-linux-gnu/libcuda.so && \
     cd /sgl-kernel && \
     ${PYTHON_ROOT_PATH}/bin/python setup.py bdist_wheel