|
|
@@ -1980,6 +1980,9 @@ extern "C" {
|
|
|
typedef void (*ggml_custom2_op_t)(struct ggml_tensor * dst , const struct ggml_tensor * a, const struct ggml_tensor * b, int ith, int nth, void * userdata);
|
|
|
typedef void (*ggml_custom3_op_t)(struct ggml_tensor * dst , const struct ggml_tensor * a, const struct ggml_tensor * b, const struct ggml_tensor * c, int ith, int nth, void * userdata);
|
|
|
|
|
|
+#define GGML_N_TASKS_MAX (-1)
|
|
|
+ // n_tasks == GGML_N_TASKS_MAX means to use max number of tasks
|
|
|
+
|
|
|
GGML_API struct ggml_tensor * ggml_map_custom1(
|
|
|
struct ggml_context * ctx,
|
|
|
struct ggml_tensor * a,
|