From 16af6f5c42cbb50201f3a2db2ba159de83600b4c Mon Sep 17 00:00:00 2001
From: TanGuofu <267266206@qq.com>
Date: Tue, 28 Mar 2017 20:03:55 +0800
Subject: [PATCH] Update mshadow.mk to support custom CUDNN_PATH

In Mxnet there is USE_CUDNN config in make/config.mk , but there is not custom cudnn_path, so this PR add a config in mshadow.mk
---
 make/mshadow.mk | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/make/mshadow.mk b/make/mshadow.mk
index 05986668..cb39b07b 100644
--- a/make/mshadow.mk
+++ b/make/mshadow.mk
@@ -33,6 +33,11 @@ ifneq ($(USE_CUDA_PATH), NONE)
 	MSHADOW_LDFLAGS += -L$(USE_CUDA_PATH)/lib64 -L$(USE_CUDA_PATH)/lib
 endif
 
+ifneq ($(USE_CUDNN_PATH), NONE)
+       MSHADOW_CFLAGS += -I$(USE_CUDNN_PATH)/include
+       MSHADOW_LDFLAGS += -L$(USE_CUDNN_PATH)/lib64 -L$(USE_CUDNN_PATH)/lib
+endif
+
 ifeq ($(USE_BLAS), mkl)
 ifneq ($(USE_INTEL_PATH), NONE)
 	UNAME_S := $(shell uname -s)