Merge pull request apache#98 from javelinjs/fix-tensor-container-release

Make it able to set ptrs to null when Release fails.
stefanhenneking · Feb 5, 2016 · 62ca0b7 · 62ca0b7
2 parents 1286bea + 6a9d036
commit 62ca0b7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mshadow/tensor_container.h b/mshadow/tensor_container.h
@@ -171,12 +171,17 @@ class TensorContainer: public Tensor<Device, dimension, DType> {
    */
   inline void Release(void) {
     if (data_.dptr_ != NULL) {
-      mshadow::FreeSpace(&data_);
-      this->dptr_ = data_.dptr_ = NULL;
       this->shape_[0] = 0;
       this->stride_ = 0;
       this->data_.stride_ = 0;
       this->data_.shape_[0] = 0;
+      try {
+        mshadow::FreeSpace(&data_);
+      } catch (const dmlc::Error &e) {
+        this->dptr_ = data_.dptr_ = NULL;
+        throw e;
+      }
+      this->dptr_ = data_.dptr_ = NULL;
     }
   }