From 74bf6bad606a8e2130e0387be595e999d00b3ab5 Mon Sep 17 00:00:00 2001 From: brodyh Date: Fri, 13 Feb 2015 20:41:14 -0800 Subject: [PATCH] Fixed an issue with MPI on unit tests --- src/caffe/common.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/caffe/common.cpp b/src/caffe/common.cpp index d0abac5ede6..02ad2f32502 100644 --- a/src/caffe/common.cpp +++ b/src/caffe/common.cpp @@ -92,7 +92,8 @@ void* Caffe::RNG::generator() { Caffe::Caffe() : cublas_handle_(NULL), curand_generator_(NULL), random_generator_(), - mode_(Caffe::CPU), phase_(Caffe::TRAIN), device_state_(Caffe::MUTABLE) { + mode_(Caffe::CPU), phase_(Caffe::TRAIN), device_state_(Caffe::MUTABLE), + mpi_(new MPILocal()) { // Try to create a cublas handler, and report an error if failed (but we will // keep the program running as one might just want to run CPU code). if (cublasCreate(&cublas_handle_) != CUBLAS_STATUS_SUCCESS) {