File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -261,6 +261,7 @@ else()
261
261
endif ()
262
262
set (MSCCLPP_USE_CUDA ON )
263
263
set (MSCCLPP_BYPASS_GPU_CHECK ON )
264
+ set (MSCCLPP_BUILD_TESTS OFF )
264
265
add_subdirectory (${repo-mscclpp_SOURCE_DIR} )
265
266
target_link_libraries (common_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda cublas cublasLt mscclpp_static )
266
267
Original file line number Diff line number Diff line change @@ -50,6 +50,9 @@ docker run --rm \
50
50
which cmake
51
51
cmake --version
52
52
53
+ yum install numactl-devel -y && \
54
+ yum install libibverbs -y && \
55
+ ln -sv /usr/lib64/libibverbs.so.1 /usr/lib64/libibverbs.so && \
53
56
${PYTHON_ROOT_PATH} /bin/${TORCH_INSTALL} && \
54
57
${PYTHON_ROOT_PATH} /bin/pip install --no-cache-dir ninja setuptools==75.0.0 wheel==0.41.0 numpy uv scikit-build-core && \
55
58
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0+PTX' && \
You can’t perform that action at this time.
0 commit comments