fix metal related CI stuff
This commit is contained in:
parent
f9212865a9
commit
36c0df7904
|
|
@ -120,7 +120,6 @@ void load_model(test_model & model, int ic, int oc, int iw, int ih, int id,
|
||||||
#ifdef GGML_USE_METAL
|
#ifdef GGML_USE_METAL
|
||||||
if (use_gpu) {
|
if (use_gpu) {
|
||||||
fprintf(stderr, "%s: using Metal backend\n", __func__);
|
fprintf(stderr, "%s: using Metal backend\n", __func__);
|
||||||
ggml_backend_metal_log_set_callback(ggml_log_callback_default, nullptr);
|
|
||||||
model.backend = ggml_backend_metal_init();
|
model.backend = ggml_backend_metal_init();
|
||||||
if (!model.backend) {
|
if (!model.backend) {
|
||||||
fprintf(stderr, "%s: ggml_backend_metal_init() failed\n", __func__);
|
fprintf(stderr, "%s: ggml_backend_metal_init() failed\n", __func__);
|
||||||
|
|
@ -310,13 +309,6 @@ std::vector<float> compute_graph(const test_model & model, ggml_gallocr_t allocr
|
||||||
ggml_backend_cpu_set_n_threads(model.backend, n_threads);
|
ggml_backend_cpu_set_n_threads(model.backend, n_threads);
|
||||||
}
|
}
|
||||||
|
|
||||||
#ifdef GGML_USE_METAL
|
|
||||||
if (ggml_backend_is_metal(model.backend)) {
|
|
||||||
ggml_backend_metal_set_n_cb(model.backend, n_threads);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
ggml_backend_graph_compute(model.backend, gf);
|
ggml_backend_graph_compute(model.backend, gf);
|
||||||
|
|
||||||
ggml_backend_synchronize(model.backend);
|
ggml_backend_synchronize(model.backend);
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue