Ameya Keshava Mallya
42d84317cf
git-subtree-dir: projects/rccl git-subtree-mainline:3fd8a0d393git-subtree-split:1f2f9f33ba
27 lines
536 B
Makefile
27 lines
536 B
Makefile
|
|
ROCM_PATH ?= /opt/rocm
|
|
CUDA_PATH ?= /usr/local/cuda
|
|
|
|
HIPCC=$(ROCM_PATH)/bin/hipcc
|
|
NVCC=$(CUDA_PATH)/bin/nvcc
|
|
|
|
# Compile TransferBenchCuda if nvcc detected
|
|
ifeq ("$(shell test -e $(NVCC) && echo found)", "found")
|
|
EXE=./EmptyKernelTestCuda
|
|
else
|
|
EXE=./EmptyKernelTest
|
|
endif
|
|
|
|
all: $(EXE)
|
|
|
|
./EmptyKernelTest: EmptyKernelTest.cpp
|
|
$(HIPCC) EmptyKernelTest.cpp -o EmptyKernelTest
|
|
|
|
./EmptyKernelTestCuda: EmptyKernelTest.cpp
|
|
$(NVCC) EmptyKernelTest.cpp -x cu -o EmptyKernelTestCuda
|
|
|
|
|
|
clean:
|
|
rm -f ./EmptyKernelTest ./EmptyKernelTestCuda
|
|
|