fix deadlock for cases where cgraph.n_nodes == 1

and fix --poll case

Author: fmz

ggml/src/ggml.c

@@ -19190,7 +19190,14 @@ static thread_ret_t ggml_graph_compute_thread(void * data) {
         }
     }
 
+    if (cgraph->n_nodes == 1) {
+        // We need a barrier before disabling new_work in case we have a trivial graph
+        ggml_barrier(state->threadpool);
+    }
+
     if (!state->threadpool->disposable && state->ith == 0) {
+        // Don't need a lock, because there is a barrier after this, and only after that
+        // do the secondary threads go into standby
         state->threadpool->new_work = false;
     }
 
@@ -19432,6 +19439,8 @@ enum ggml_status ggml_graph_compute(struct ggml_cgraph * cgraph, struct ggml_cpl
             threadpool->new_work = true;
             ggml_cond_broadcast(&threadpool->cond);
             ggml_mutex_unlock(&threadpool->mutex);
+        } else {
+            threadpool->new_work = true;
         }
     }
     // this is a work thread too