|
|
@@ -473,8 +473,8 @@ extern "C" {
|
|
|
int n_dims;
|
|
|
int64_t ne[GGML_MAX_DIMS]; // number of elements
|
|
|
size_t nb[GGML_MAX_DIMS]; // stride in bytes:
|
|
|
- // nb[0] = sizeof(type)
|
|
|
- // nb[1] = nb[0] * ne[0] + padding
|
|
|
+ // nb[0] = ggml_type_size(type)
|
|
|
+ // nb[1] = nb[0] * (ne[0] / ggml_blck_size(type)) + padding
|
|
|
// nb[i] = nb[i-1] * ne[i-1]
|
|
|
|
|
|
// compute data
|