add Tensor::is_contiguous

`Tensor::is_contiguous`: returns true if this tensor is stored contiguously in memory
rustformers · Aug 18, 2023 · 8bab9c5 · 8bab9c5
1 parent 129b84a
commit 8bab9c5
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 {