Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
Makefile
View file @ 42b7d85b
...@@ -16,8 +16,8 @@ GPU_TARGETS ?= native ...@@ -16,8 +16,8 @@ GPU_TARGETS ?= native
DEBUG ?= 0 DEBUG ?= 0
ifeq ($(filter clean,$(MAKECMDGOALS)),) ifeq ($(filter clean,$(MAKECMDGOALS)),)
# Compile TransferBenchCuda if nvcc detected # Compile TransferBenchCuda if nvidia-smi returns successfully and nvcc detected
ifeq ("$(shell test -e $(NVCC) && echo found)", "found") ifeq ("$(shell nvidia-smi > /dev/null 2>&1 && test -e $(NVCC) && echo found)", "found")
EXE=TransferBenchCuda EXE=TransferBenchCuda
CXX=$(NVCC) CXX=$(NVCC)
else else
...@@ -47,6 +47,7 @@ ifeq ($(filter clean,$(MAKECMDGOALS)),) ...@@ -47,6 +47,7 @@ ifeq ($(filter clean,$(MAKECMDGOALS)),)
LDFLAGS += -lpthread LDFLAGS += -lpthread
NIC_ENABLED = 0
# Compile RDMA executor if # Compile RDMA executor if
# 1) DISABLE_NIC_EXEC is not set to 1 # 1) DISABLE_NIC_EXEC is not set to 1
# 2) IBVerbs is found in the Dynamic Linker cache # 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