Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "85526e34a18d806d9447f8f21142cab2e07a0229"
Unverified Commit 9d54c6e6 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

feat: remove sm75 (#9207)

parent 1f9d65f5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
sgl-kernel/CMakeLists.txt
View file @ 9d54c6e6
...@@ -170,7 +170,6 @@ option(SGL_KERNEL_ENABLE_FA3 "Enable FA3" OFF) ...@@ -170,7 +170,6 @@ option(SGL_KERNEL_ENABLE_FA3 "Enable FA3" OFF)
if (ENABLE_BELOW_SM90) if (ENABLE_BELOW_SM90)
list(APPEND SGL_KERNEL_CUDA_FLAGS list(APPEND SGL_KERNEL_CUDA_FLAGS
"-gencode=arch=compute_75,code=sm_75"
"-gencode=arch=compute_80,code=sm_80" "-gencode=arch=compute_80,code=sm_80"
"-gencode=arch=compute_89,code=sm_89" "-gencode=arch=compute_89,code=sm_89"
) )
......
sgl-kernel/build.sh
View file @ 9d54c6e6
...@@ -63,7 +63,7 @@ docker run --rm \ ...@@ -63,7 +63,7 @@ docker run --rm \
ln -sv /usr/lib64/libibverbs.so.1 /usr/lib64/libibverbs.so && \ ln -sv /usr/lib64/libibverbs.so.1 /usr/lib64/libibverbs.so && \
${PYTHON_ROOT_PATH}/bin/${TORCH_INSTALL} && \ ${PYTHON_ROOT_PATH}/bin/${TORCH_INSTALL} && \
${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja setuptools==75.0.0 wheel==0.41.0 numpy uv scikit-build-core && \ ${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja setuptools==75.0.0 wheel==0.41.0 numpy uv scikit-build-core && \
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \ export TORCH_CUDA_ARCH_LIST='8.0 8.9 9.0+PTX' && \
export CUDA_VERSION=${CUDA_VERSION} && \ export CUDA_VERSION=${CUDA_VERSION} && \
mkdir -p /usr/lib/${ARCH}-linux-gnu/ && \ mkdir -p /usr/lib/${ARCH}-linux-gnu/ && \
ln -s /usr/local/cuda-${CUDA_VERSION}/targets/${LIBCUDA_ARCH}-linux/lib/stubs/libcuda.so /usr/lib/${ARCH}-linux-gnu/libcuda.so && \ ln -s /usr/local/cuda-${CUDA_VERSION}/targets/${LIBCUDA_ARCH}-linux/lib/stubs/libcuda.so /usr/lib/${ARCH}-linux-gnu/libcuda.so && \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment