SplitLayer problem about fb-caffe-exts HOT 6 CLOSED

--- a/caffe/caffe-trunk/src/caffe/layers/split_layer.cpp
+++ b/caffe/caffe-trunk/src/caffe/layers/split_layer.cpp
@@ -16,8 +16,6 @@
     // the backward pass.  (Technically, it should be possible to share the diff
     // blob of the first split output with the input, but this seems to cause
     // some strange effects in practice...)
-    CHECK_NE(top[i], bottom[0]) << this->type() << " Layer does not "
-        "allow in-place computation.";
     top[i]->ReshapeLike(*bottom[0]);
     CHECK_EQ(count_, top[i]->count());
   }
@@ -36,13 +34,17 @@
       const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {
   if (!propagate_down[0]) { return; }
   if (top.size() == 1) {
+    CHECK_NE(top[0], bottom[0]) << this->type() << " Layer does not "
+        "allow in-place computation.";
     caffe_copy(count_, top[0]->cpu_diff(), bottom[0]->mutable_cpu_diff());
     return;
   }
   caffe_add(count_, top[0]->cpu_diff(), top[1]->cpu_diff(),
             bottom[0]->mutable_cpu_diff());
   // Add remaining top blob diffs.
   for (int i = 2; i < top.size(); ++i) {
+    CHECK_NE(top[i], bottom[0]) << this->type() << " Layer does not "
+        "allow in-place computation.";
     const Dtype* top_diff = top[i]->cpu_diff();
     Dtype* bottom_diff = bottom[0]->mutable_cpu_diff();
     caffe_axpy(count_, Dtype(1.), top_diff, bottom_diff);