Add buffer sync methods hannk::Tensor class

Add few methods for GPU memory interaction.
halide · Feb 6, 2023 · a6c65ed · a6c65ed
1 parent 91f3ac0
commit a6c65ed
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/apps/hannk/interpreter/tensor.h b/apps/hannk/interpreter/tensor.h
@@ -231,6 +231,20 @@ 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 {