Merge pull request #410 from nerdypepper/add-is-contiguous

add `Tensor::is_contiguous`
rustformers · Aug 18, 2023 · 56e4a35 · 56e4a35
2 parents 129b84a + 8bab9c5
commit 56e4a35
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/ggml/src/tensor.rs b/crates/ggml/src/tensor.rs
@@ -220,6 +220,11 @@ impl Tensor {
             sys::opencl::ggml_cl_free_data(self.ptr.as_ptr());
         }
     }
+
+    /// Returns true if this tensor is stored contiguously in memory
+    pub fn is_contiguous(&self) -> bool {
+        unsafe { sys::ggml_is_contiguous(self.ptr.as_ptr()) }
+    }
 }
 impl Tensor {
     fn with_alive_ctx<U>(&self, mut f: impl FnMut() -> U) -> U {