paths to include for cuda and ze

Author: JuliaRS

src_c/IMB_gpu_common.h

@@ -64,10 +64,10 @@ For more documentation than found here, see
 #include <stdlib.h>
 #include <dlfcn.h>
 #include "IMB_comm_info.h"
-#if defined(ZE_INCLUDE_DIR)
+#ifdef ZE_INCLUDE_DIR
 #include "IMB_ze.h"
 #endif
-#if defined(CUDA_INCLUDE_DIR)
+#ifdef CUDA_INCLUDE_DIR
 #include "IMB_cuda.h"
 #endif
 

src_cpp/MPI1/Makefile.MPI1.mk

@@ -93,12 +93,12 @@ override CPPFLAGS += -DGPU_ENABLE -ldl
 ifdef CUDA_INCLUDE_DIR
 override C_SRC += $(C_SRC_DIR)/IMB_cuda.c \
 $(C_SRC_DIR)/IMB_cuda_api.c
-override CPPFLAGS += -I${CUDA_INCLUDE_DIR}
+override CPPFLAGS += -I${CUDA_INCLUDE_DIR} -DCUDA_INCLUDE_DIR
 endif
 ifdef ZE_INCLUDE_DIR
 override C_SRC += $(C_SRC_DIR)/IMB_ze.c \
 $(C_SRC_DIR)/IMB_ze_api.c
-override CPPFLAGS += -I${ZE_INCLUDE_DIR}
+override CPPFLAGS += -I${ZE_INCLUDE_DIR} -DZE_INCLUDE_DIR
 endif
 ifndef CUDA_INCLUDE_DIR
 ifndef ZE_INCLUDE_DIR