Skip to content

Fix ggml_sizeof_tensor functions #3

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed

Fix ggml_sizeof_tensor functions #3

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions src/ggml.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4075,17 +4075,17 @@ size_t ggml_sizeof_tensor_1d(enum ggml_type type, int64_t ne0) {
}

size_t ggml_sizeof_tensor_2d(enum ggml_type type, int64_t ne0, int64_t ne1) {
auto ne = {ne0, ne1};
int64_t ne[2] = {ne0, ne1};
return ggml_tensor_overhead() + ggml_sizeof_tensor_data_impl(type, 2, &ne);
}

size_t ggml_sizeof_tensor_3d(enum ggml_type type, int64_t ne0, int64_t ne1, int64_t ne2) {
auto ne = {ne0, ne1, ne2};
int64_t ne[3] = {ne0, ne1, ne2};
return ggml_tensor_overhead() + ggml_sizeof_tensor_data_impl(type, 3, &ne);
}

size_t ggml_sizeof_tensor_4d(enum ggml_type type, int64_t ne0, int64_t ne1, int64_t ne2, int64_t ne3) {
auto ne = {ne0, ne1, ne2, ne3};
int64_t ne[4] = {ne0, ne1, ne2, ne3};
return ggml_tensor_overhead() + ggml_sizeof_tensor_data_impl(type, 4, &ne);
}

Expand Down