fix: set metal_n_cb in test

abetlen · Feb 9, 2024 · bb58e86 · bb58e86
1 parent ff00b0b
commit bb58e86
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_ggml_backends.py b/tests/test_ggml_backends.py
@@ -8,7 +8,9 @@
 def test_ggml_backend():
     def get_backend():
         if ggml.ggml_cpu_has_metal():
-            return ggml.ggml_backend_metal_init()
+            backend = ggml.ggml_backend_metal_init()
+            ggml.ggml_backend_metal_set_n_cb(backend, 1)
+            return backend
         elif ggml.ggml_cpu_has_cublas():
             return ggml.ggml_backend_cuda_init()
         elif ggml.ggml_cpu_has_vulkan():