diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/mxnet/makefile-cudnn-path-1.6.patch')
| -rw-r--r-- | var/spack/repos/builtin/packages/mxnet/makefile-cudnn-path-1.6.patch | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/mxnet/makefile-cudnn-path-1.6.patch b/var/spack/repos/builtin/packages/mxnet/makefile-cudnn-path-1.6.patch new file mode 100644 index 0000000000..8ae19440ad --- /dev/null +++ b/var/spack/repos/builtin/packages/mxnet/makefile-cudnn-path-1.6.patch @@ -0,0 +1,13 @@ +diff --git a/Makefile b/Makefile +index 4746cc434..e738fc53f 100644 +--- a/Makefile ++++ b/Makefile +@@ -509,7 +509,7 @@ LIB_DEP += $(DMLC_CORE)/libdmlc.a $(NNVM_PATH)/lib/libnnvm.a + ALL_DEP = $(OBJ) $(EXTRA_OBJ) $(PLUGIN_OBJ) $(LIB_DEP) + + ifeq ($(USE_CUDA), 1) +- CFLAGS += -I$(ROOTDIR)/3rdparty/nvidia_cub ++ CFLAGS += -I$(CUB_INCLUDE) -I$(CUDNN_PATH)/include + ALL_DEP += $(CUOBJ) $(EXTRA_CUOBJ) $(PLUGIN_CUOBJ) + LDFLAGS += -lcufft + ifeq ($(ENABLE_CUDA_RTC), 1) |