update

Author: jwjohns

src/llama-graph.cpp

@@ -238,6 +238,11 @@ void llm_graph_input_rs::set_input(const llama_ubatch * ubatch) {
     const int64_t n_rs = mctx->get_n_rs();
 
     if (s_copy) {
+        // Check if buffer was allocated - skip if not
+        if (s_copy->buffer == nullptr) {
+            fprintf(stderr, "[DEBUG] RS s_copy buffer is NULL, skipping copy operations\n");
+            return;
+        }
         GGML_ASSERT(ggml_backend_buffer_is_host(s_copy->buffer));
         int32_t * data = (int32_t *) s_copy->data;
 
@@ -392,8 +397,17 @@ void llm_graph_input_attn_cross::set_input(const llama_ubatch * ubatch) {
 }
 
 void llm_graph_input_mem_hybrid::set_input(const llama_ubatch * ubatch) {
-    inp_attn->set_input(ubatch);
-    inp_rs->set_input(ubatch);
+    fprintf(stderr, "[DEBUG] hybrid set_input: inp_attn=%p, inp_rs=%p\n", (void*)inp_attn.get(), (void*)inp_rs.get());
+    if (inp_attn) {
+        inp_attn->set_input(ubatch);
+    } else {
+        fprintf(stderr, "[ERROR] inp_attn is null!\n");
+    }
+    if (inp_rs) {
+        inp_rs->set_input(ubatch);
+    } else {
+        fprintf(stderr, "[ERROR] inp_rs is null!\n");
+    }
 }
 
 //