Skip to content

Commit 7b3f95e

Browse files
committed
fix nv link
1 parent 9c9d766 commit 7b3f95e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mooncake-transfer-engine/nvlink-allocator/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ CPP_FILE=$(dirname $(readlink -f $0))/nvlink_allocator.cpp # get cpp file path,
2020

2121
# Use nvcc or g++ based on flag
2222
if [ "$USE_NVCC" = true ]; then
23-
nvcc "$CPP_FILE" -o "$OUTPUT_DIR/nvlink_allocator.so" -shared -Xcompiler -fPIC -lcuda -I/usr/local/cuda/include
23+
nvcc "$CPP_FILE" -o "$OUTPUT_DIR/nvlink_allocator.so" -shared -Xcompiler -fPIC -I/usr/local/cuda/include
2424
else
2525
g++ "$CPP_FILE" -o "$OUTPUT_DIR/nvlink_allocator.so" --shared -fPIC -lcuda -I/usr/local/cuda/include
2626
fi

0 commit comments

Comments
 (0)