Diego Devesa commited on
Commit
c4d59b9
·
1 Parent(s): 7e2b09b

metal : fix from ptr buffer name (llama/10189)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-metal.m +3 -2
ggml/src/ggml-metal.m CHANGED
@@ -4072,7 +4072,7 @@ static ggml_backend_buffer_t ggml_backend_metal_device_buffer_from_ptr(ggml_back
4072
  }
4073
  }
4074
 
4075
- return ggml_backend_buffer_init(ggml_backend_metal_buffer_type(), ggml_backend_metal_buffer_i, ctx, size);
4076
  }
4077
 
4078
  static bool ggml_backend_metal_device_supports_op(ggml_backend_dev_t dev, const struct ggml_tensor * op) {
@@ -4082,7 +4082,8 @@ static bool ggml_backend_metal_device_supports_op(ggml_backend_dev_t dev, const
4082
  }
4083
 
4084
  static bool ggml_backend_metal_device_supports_buft(ggml_backend_dev_t dev, ggml_backend_buffer_type_t buft) {
4085
- return buft->iface.get_name == ggml_backend_metal_buffer_type_get_name;
 
4086
 
4087
  UNUSED(dev);
4088
  }
 
4072
  }
4073
  }
4074
 
4075
+ return ggml_backend_buffer_init(ggml_backend_metal_buffer_from_ptr_type(), ggml_backend_metal_buffer_i, ctx, size);
4076
  }
4077
 
4078
  static bool ggml_backend_metal_device_supports_op(ggml_backend_dev_t dev, const struct ggml_tensor * op) {
 
4082
  }
4083
 
4084
  static bool ggml_backend_metal_device_supports_buft(ggml_backend_dev_t dev, ggml_backend_buffer_type_t buft) {
4085
+ return buft->iface.get_name == ggml_backend_metal_buffer_type_get_name ||
4086
+ buft->iface.get_name == ggml_backend_metal_buffer_from_ptr_type_get_name;
4087
 
4088
  UNUSED(dev);
4089
  }