Skip to content

Commit 258e7b0

Browse files
committed
fix sgl-kernel build issue
1 parent a688d4e commit 258e7b0

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

sgl-kernel/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,7 @@ else()
261261
endif()
262262
set(MSCCLPP_USE_CUDA ON)
263263
set(MSCCLPP_BYPASS_GPU_CHECK ON)
264+
set(MSCCLPP_BUILD_TESTS OFF)
264265
add_subdirectory(${repo-mscclpp_SOURCE_DIR})
265266
target_link_libraries(common_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda cublas cublasLt mscclpp_static)
266267

sgl-kernel/build.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,9 @@ docker run --rm \
5050
which cmake
5151
cmake --version
5252
53+
yum install numactl-devel -y && \
54+
yum install libibverbs -y && \
55+
ln -sv /usr/lib64/libibverbs.so.1 /usr/lib64/libibverbs.so && \
5356
${PYTHON_ROOT_PATH}/bin/${TORCH_INSTALL} && \
5457
${PYTHON_ROOT_PATH}/bin/pip install --no-cache-dir ninja setuptools==75.0.0 wheel==0.41.0 numpy uv scikit-build-core && \
5558
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \

0 commit comments

Comments
 (0)