ggml: loop over all elements in view and src tensor

danbev · Nov 30, 2024 · 166b777 · 166b777
1 parent 5869e0d
commit 166b777
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/fundamentals/ggml/src/views.c b/fundamentals/ggml/src/views.c
@@ -56,6 +56,7 @@ int main(int argc, char **argv) {
     ggml_backend_graph_compute(backend, gf);
 
     {
+        printf("Loop over all dimensions in view:\n");
         float buf[6];
         ggml_backend_tensor_get(view, buf, 0, sizeof(buf));
         for (int row = 0; row < view->ne[1]; row++) {
@@ -67,6 +68,16 @@ int main(int argc, char **argv) {
     }
 
     {
+        printf("Loop over all elements in view:\n");
+        float buf[6];
+        ggml_backend_tensor_get(view, buf, 0, sizeof(buf));
+        for (int i = 0; i < ggml_nelements(view); i++) {
+            printf("buf[%d] = %f\n", i, buf[i]);
+        }
+    }
+
+    {
+        printf("Loop over all elements in src tensor:\n");
         float buf[6];
         ggml_backend_tensor_get(x, buf, 0, sizeof(buf));
         for (int i = 0; i < x->ne[0]; i++) {