task-debuginfo-fix: - Fix to crash of debug_info for MKL2017

HulkMaker · Dec 3, 2016 · 1ed431b · 1ed431b
2 parents 6368fcc + 08fceec
commit 1ed431b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/caffe/blob.cpp b/src/caffe/blob.cpp
@@ -304,6 +304,9 @@ template <typename Dtype>
 Dtype Blob<Dtype>::asum_data() const {
   if (!data_) { return 0; }
   switch (data_->head()) {
+  case SyncedMemory::SYNCED_PRV:
+  case SyncedMemory::HEAD_AT_PRV:
+    return caffe_cpu_asum( prv_data_count(), prv_data());
   case SyncedMemory::HEAD_AT_CPU:
     return caffe_cpu_asum(count_, cpu_data());
   case SyncedMemory::HEAD_AT_GPU: