|
21 | 21 | #define DEBUG_POS 5
|
22 | 22 |
|
23 | 23 | static void print_debug_tensor(struct ggml_tensor * t, bool with_data = true) {
|
24 |
| - printf("%s: %s (%s): [%ld, %ld]\n", __func__, t->name, ggml_type_name(t->type), t->ne[0], t->ne[1]); |
| 24 | + printf("%s: %s (%s): [%ld, %ld]\n", __func__, t->name, ggml_type_name(t->type), (size_t) t->ne[0], (size_t) t->ne[1]); |
25 | 25 | if (!with_data) return;
|
26 | 26 | printf("%s: %s[0] = [", __func__, t->name);
|
27 | 27 | for (size_t i = 0; i <= DEBUG_POS; i++) {
|
@@ -73,7 +73,6 @@ struct pca_model {
|
73 | 73 |
|
74 | 74 | #ifdef GGML_USE_METAL
|
75 | 75 | fprintf(stderr, "%s: using Metal backend\n", __func__);
|
76 |
| - ggml_backend_metal_log_set_callback(ggml_log_callback_default, nullptr); |
77 | 76 | backend = ggml_backend_metal_init();
|
78 | 77 | if (!backend) {
|
79 | 78 | fprintf(stderr, "%s: ggml_backend_metal_init() failed\n", __func__);
|
@@ -185,7 +184,7 @@ static struct ggml_cgraph * build_graph_piter(
|
185 | 184 | ggml_set_name(b_tensor, ("b_tensor_norm_" + std::to_string(i)).c_str());
|
186 | 185 |
|
187 | 186 | // calculate distance(new eigenvector - old eigenvector)
|
188 |
| - struct ggml_tensor * new_sub_old = ggml_sub(ctx0, old_eigen, b_tensor); |
| 187 | + struct ggml_tensor * new_sub_old = ggml_add(ctx0, old_eigen, ggml_scale(ctx0, b_tensor, -1)); |
189 | 188 | distance = ggml_sqrt_inplace(ctx0,
|
190 | 189 | ggml_sum_rows(ctx0, ggml_sqr_inplace(ctx0, new_sub_old)));
|
191 | 190 | ggml_set_name(distance, ("distance_" + std::to_string(i)).c_str());
|
|
0 commit comments