21 lines
456 B
Bash
21 lines
456 B
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
# Detect architecture
|
|
ARCH=$(uname -m)
|
|
|
|
# Default paths (adjust as needed)
|
|
CUDA_PATH="${CUDA_PATH:-/usr/local/cuda}"
|
|
CUDSS_PATH="${CUDSS_PATH:-/usr/local/cudss}"
|
|
|
|
nvcc -o cudss_test cudss_test.cu \
|
|
-I"${CUDA_PATH}/include" \
|
|
-I"${CUDSS_PATH}/include" \
|
|
-L"${CUDA_PATH}/lib64" \
|
|
-L"${CUDSS_PATH}/lib" \
|
|
-lcudss -lcudart \
|
|
-ccbin clang++ \
|
|
-Xcompiler "-Wall -O2"
|
|
|
|
echo "Build successful: cudss_test (${ARCH})"
|