Skip to content

Commit 95d4a03

Browse files
committed
rename ggml_custom -> ggml_custom_4d
1 parent 62b4780 commit 95d4a03

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

include/ggml.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1968,7 +1968,7 @@ extern "C" {
19681968

19691969
typedef void (*ggml_custom_op_t)(struct ggml_tensor * dst , int ith, int nth, void * userdata);
19701970

1971-
GGML_API struct ggml_tensor * ggml_custom(
1971+
GGML_API struct ggml_tensor * ggml_custom_4d(
19721972
struct ggml_context * ctx,
19731973
enum ggml_type type,
19741974
int64_t ne0,

src/ggml.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4970,7 +4970,7 @@ struct ggml_tensor * ggml_map_custom3_inplace(
49704970
return ggml_map_custom3_impl(ctx, a, b, c, fun, n_tasks, userdata, true);
49714971
}
49724972

4973-
struct ggml_tensor * ggml_custom(
4973+
struct ggml_tensor * ggml_custom_4d(
49744974
struct ggml_context * ctx,
49754975
enum ggml_type type,
49764976
int64_t ne0,

tests/test-customop.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ int main(int argc, const char** argv) {
280280
t1, t2, t3, t4, t5,
281281
};
282282

283-
struct ggml_tensor * m4 = ggml_custom(ctx, GGML_TYPE_I32, 10, 2, 1, 1, args, sizeof(args)/sizeof(args[0]), custom, GGML_N_TASKS_MAX, g_userdata);
283+
struct ggml_tensor * m4 = ggml_custom_4d(ctx, GGML_TYPE_I32, 10, 2, 1, 1, args, sizeof(args)/sizeof(args[0]), custom, GGML_N_TASKS_MAX, g_userdata);
284284

285285
struct ggml_cgraph * graph = ggml_new_graph(ctx);
286286
ggml_build_forward_expand(graph, m4);

0 commit comments

Comments
 (0)