Merge pull request halide#2196 from halide/srj-buf

Buffer::copy and Buffer::make_with_shape_of need to forward optional …
liqunfu · Jul 14, 2017 · 4ffac12 · 4ffac12
2 parents cf96af8 + 2e26ce7
commit 4ffac12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/runtime/HalideBuffer.h b/src/runtime/HalideBuffer.h
@@ -963,7 +963,7 @@ class Buffer {
      * original, with holes compacted away. */
     Buffer<T, D> copy(void *(*allocate_fn)(size_t) = nullptr,
                       void (*deallocate_fn)(void *) = nullptr) const {
-        Buffer<T, D> dst = make_with_shape_of(*this);
+        Buffer<T, D> dst = make_with_shape_of(*this, allocate_fn, deallocate_fn);
         dst.copy_from(*this);
         return dst;
     }
@@ -1420,7 +1420,7 @@ class Buffer {
         }
 
         Buffer<T, D> dst(nullptr, src.dimensions(), shape);
-        dst.allocate();
+        dst.allocate(allocate_fn, deallocate_fn);
 
         return dst;
     }