@@ -30,32 +30,30 @@ int main(int argc, char **argv) {
30
30
struct ggml_context * ctx = ggml_init (params );
31
31
struct ggml_tensor * x = ggml_new_tensor_1d (ctx , GGML_TYPE_F32 , 10 );
32
32
ggml_set_name (x , "x" );
33
- printf ("x backend type (0=CPU, 10=GPU): %d\n" , x -> backend );
34
33
if (x -> buffer == NULL ) {
35
34
printf ("x backend buffer is NULL\n" );
36
35
} else {
37
36
printf ("x backend buffer: %s\n" , ggml_backend_buffer_name (x -> buffer ));
38
37
}
39
38
40
39
// The following will call ggml_backend_registry_init
41
- size_t count = ggml_backend_reg_get_count ();
40
+ size_t count = ggml_backend_reg_count ();
42
41
printf ("Number of backends registered: %ld\n" , count );
43
42
for (size_t i = 0 ; i < count ; i ++ ) {
44
- printf ("backend %ld name: %s\n" , i , ggml_backend_reg_get_name ( i ));
43
+ printf ("backend %ld name: %s\n" , i , ggml_backend_dev_name ( ggml_backend_dev_get ( i ) ));
45
44
}
46
45
47
- ggml_backend_t cpu_backend = ggml_backend_reg_init_backend_from_str ("CPU" );
46
+ ggml_backend_t cpu_backend = ggml_backend_init_by_name ("CPU" , NULL );
48
47
if (cpu_backend != NULL ) {
49
48
ggml_backend_buffer_t buffer = ggml_backend_alloc_buffer (cpu_backend , 10 * 4 );
50
49
print_backend_info (buffer , ctx );
51
50
ggml_backend_free (cpu_backend );
52
51
}
53
52
54
- ggml_backend_t cuda_backend = ggml_backend_reg_init_backend_from_str ("CUDA0" );
53
+ ggml_backend_t cuda_backend = ggml_backend_init_by_name ("CUDA0" , NULL );
55
54
if (cuda_backend != NULL ) {
56
55
ggml_backend_buffer_t buffer = ggml_backend_alloc_buffer (cuda_backend , 10 * 4 );
57
56
print_backend_info (buffer , ctx );
58
- printf ("x backend type (%d=CPU, %d=GPU): %d\n" , GGML_BACKEND_TYPE_CPU , GGML_BACKEND_TYPE_GPU , x -> backend );
59
57
if (x -> buffer != NULL ) {
60
58
printf ("x backend buffer: %s\n" , ggml_backend_buffer_name (x -> buffer ));
61
59
}
@@ -67,7 +65,6 @@ int main(int argc, char **argv) {
67
65
68
66
// The following will copy the data from the host to the device.
69
67
ggml_backend_tensor_set (x , data , 0 , 10 );
70
- printf ("x backend type (%d=CPU, %d=GPU): %d\n" , GGML_BACKEND_TYPE_CPU , GGML_BACKEND_TYPE_GPU , x -> backend );
71
68
ggml_backend_free (cuda_backend );
72
69
}
73
70
0 commit comments