Просмотр исходного кода

ggml: disable CUDA graphs for unsupported DUP and CONT node types (#12891)

Fixes #12798
Alan Gray 9 месяцев назад
Родитель
Сommit
307bfa253d
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      ggml/src/ggml-cuda/ggml-cuda.cu

+ 2 - 2
ggml/src/ggml-cuda/ggml-cuda.cu

@@ -2488,10 +2488,10 @@ static bool check_node_graph_compatibility_and_refresh_copy_ops(ggml_backend_cud
 #endif
         }
 
-        if (node->op == GGML_OP_MUL_MAT_ID) {
+        if (node->op == GGML_OP_MUL_MAT_ID || node->op == GGML_OP_CONT || node->op == GGML_OP_DUP) {
             use_cuda_graph = false; // This node type is not supported by CUDA graph capture
 #ifndef NDEBUG
-            GGML_LOG_DEBUG("%s: disabling CUDA graphs due to mul_mat_id\n", __func__);
+            GGML_LOG_DEBUG("%s: disabling CUDA graphs due to unsupported node type\n", __func__);
 #endif
         }