diff --git a/amalgamation/amalgamation.py b/amalgamation/amalgamation.py index da3b60ac8399..b33b81c62b4a 100644 --- 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 index d51deb285c99..ca1b581ce195 100644 --- 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"