18 lines
328 B
Bash
18 lines
328 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# CUDA installation directory
|
||
|
|
CUDA_ROOT=/usr/local/cuda
|
||
|
|
|
||
|
|
# CUDA Compiler (nvcc)
|
||
|
|
CUX=${CUDA_ROOT}/bin/nvcc
|
||
|
|
|
||
|
|
# Compile file1.cpp
|
||
|
|
g++ -c -I${CUDA_ROOT}/include -o file1.o file1.cpp
|
||
|
|
|
||
|
|
# Compile file2.cu
|
||
|
|
${CUX} -c -o file2.o file2.cu
|
||
|
|
|
||
|
|
# Link file1.o and file2.o
|
||
|
|
g++ -o main file1.o file2.o -L${CUDA_ROOT}/lib64 -lcudart
|
||
|
|
|