Fix broken amalgamation. Script was copying a declaration from a head…

…er file but missing out the definition, causing a runtime undef symbol error. (apache#7222)
Guneet-Dhillon · Jul 27, 2017 · 08b786a · 08b786a
1 parent 2d4c32f
commit 08b786a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/amalgamation/amalgamation.py b/amalgamation/amalgamation.py
@@ -8,7 +8,8 @@
     'kvstore_dist.h', 'mach/clock.h', 'mach/mach.h',
     'malloc.h', 'mkl.h', 'mkl_cblas.h', 'mkl_vsl.h', 'mkl_vsl_functions.h',
     'nvml.h', 'opencv2/opencv.hpp', 'sys/stat.h', 'sys/types.h', 'cuda.h', 'cuda_fp16.h',
-    'omp.h', 'execinfo.h', 'packet/sse-inl.h', 'emmintrin.h', 'thrust/device_vector.h'
+    'omp.h', 'execinfo.h', 'packet/sse-inl.h', 'emmintrin.h', 'thrust/device_vector.h', 
+    'cusolverDn.h'
     ]
 
 minimum = int(sys.argv[6]) if len(sys.argv) > 5 else 0

diff --git a/amalgamation/mxnet_predict0.cc b/amalgamation/mxnet_predict0.cc
@@ -26,6 +26,7 @@
 
 
 #include "src/ndarray/ndarray_function.cc"
+#include "src/ndarray/autograd.cc"
 #include "src/ndarray/ndarray.cc"
 
 #include "src/engine/engine.cc"