Unverified Commit 42b7d85b authored by Mustafa Abduljabbar's avatar Mustafa Abduljabbar Committed by GitHub
Browse files

Check nvidia-smi output instead of just searching nvcc bin in makefile (#202)

parent a824bc1b
......@@ -16,8 +16,8 @@ GPU_TARGETS ?= native
DEBUG ?= 0
ifeq ($(filter clean,$(MAKECMDGOALS)),)
# Compile TransferBenchCuda if nvcc detected
ifeq ("$(shell test -e $(NVCC) && echo found)", "found")
# Compile TransferBenchCuda if nvidia-smi returns successfully and nvcc detected
ifeq ("$(shell nvidia-smi > /dev/null 2>&1 && test -e $(NVCC) && echo found)", "found")
EXE=TransferBenchCuda
CXX=$(NVCC)
else
......@@ -47,6 +47,7 @@ ifeq ($(filter clean,$(MAKECMDGOALS)),)
LDFLAGS += -lpthread
NIC_ENABLED = 0
# Compile RDMA executor if
# 1) DISABLE_NIC_EXEC is not set to 1
# 2) IBVerbs is found in the Dynamic Linker cache
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment