浏览代码

check prologue callback arg in insert task

Andra Hugo 12 年之前
父节点
当前提交
2d32de2f22
共有 1 个文件被更改,包括 10 次插入0 次删除
  1. 10 0
      src/util/starpu_insert_task_utils.c

+ 10 - 0
src/util/starpu_insert_task_utils.c

@@ -80,6 +80,16 @@ size_t _starpu_insert_task_get_arg_size(va_list varg_list)
 			va_arg(varg_list, _starpu_callback_func_t);
 			va_arg(varg_list, void *);
 		}
+		else if (arg_type==STARPU_PROLOGUE_CALLBACK)
+		{
+			(void)va_arg(varg_list, _starpu_callback_func_t);
+		}
+		else if (arg_type==STARPU_PROLOGUE_CALLBACK_ARG)
+		{
+			va_arg(varg_list, _starpu_callback_func_t);
+			va_arg(varg_list, void *);
+		}
+
 		else if (arg_type==STARPU_CALLBACK_ARG)
 		{
 			(void)va_arg(varg_list, void *);