adding cuda support for EmptyKernelTest (#913)

This commit is contained in:
mberenjk
2023-10-11 14:11:12 -05:00
committato da GitHub
parent 01d9da8046
commit 7e2d905376
2 ha cambiato i file con 78 aggiunte e 5 eliminazioni
+22 -4
Vedi File
@@ -1,8 +1,26 @@
CXX=/opt/rocm/bin/hipcc
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
EmptyKernelTest: EmptyKernelTest.cpp
$(CXX) EmptyKernelTest.cpp -o EmptyKernelTest
clean:
rm -f ./EmptyKernelTest
rm -f ./EmptyKernelTest ./EmptyKernelTestCuda