Add buffer sync methods hannk::Tensor class

Add few methods for GPU memory interaction.
halide · Feb 6, 2023 · 44a01b9 · 44a01b9
1 parent 91f3ac0
commit 44a01b9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/apps/hannk/interpreter/tensor.h b/apps/hannk/interpreter/tensor.h
@@ -231,6 +231,12 @@ class Tensor {
     void allocate_from_heap();
     void allocate_from_arena_pointer(void *host);
 
+    // HalideBuffer methods for GPU interactions.
+    void set_host_dirty (bool dirty = true) { buffer_.set_host_dirty(dirty); }
+    void copy_to_host() { buffer_.copy_to_host(); }
+    bool device_dirty() const { return buffer_.device_dirty(); }
+    bool host_dirty() const { return buffer_.host_dirty(); }
+
     void resize_dynamic(const Box &new_shape);
 
     AliasType alias_type() const {