vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

minor cleanup

Browse source
This commit is contained in:
Jared Van Bortel 2024-01-19 15:05:01 -05:00
parent 33e8d6abe1
commit cb9ceff966
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ggml-kompute.cpp
View file

@ -1289,7 +1289,7 @@ static void ggml_vk_cpy_f16_f32(Args&&... args) {
ggml_vk_cpy<2, 4>(spirv, std::forward<Args>(args)...); ggml_vk_cpy<2, 4>(spirv, std::forward<Args>(args)...);
} }
static bool ggml_kompute_supports_op(const struct ggml_tensor * op) { static bool ggml_vk_supports_op(const struct ggml_tensor * op) {
switch (op->type) { switch (op->type) {
case GGML_TYPE_F16: case GGML_TYPE_F16:
case GGML_TYPE_F32: case GGML_TYPE_F32:
@ -1417,7 +1417,7 @@ void ggml_vk_graph_compute(struct ggml_kompute_context * ctx, struct ggml_cgraph
any_commands_recorded = true; any_commands_recorded = true;
if (!ggml_kompute_supports_op(dst)) { if (!ggml_vk_supports_op(dst)) {
fprintf(stderr, "%s: error: unsupported op '%s'\n", __func__, ggml_op_desc(dst)); fprintf(stderr, "%s: error: unsupported op '%s'\n", __func__, ggml_op_desc(dst));
GGML_ASSERT(!"unsupported op"); GGML_ASSERT(!"unsupported op");
} }
@ -1828,7 +1828,7 @@ static bool ggml_backend_kompute_graph_compute(ggml_backend_t backend, struct gg
static bool ggml_backend_kompute_supports_op(ggml_backend_t backend, const struct ggml_tensor * op) { static bool ggml_backend_kompute_supports_op(ggml_backend_t backend, const struct ggml_tensor * op) {
GGML_UNUSED(backend); GGML_UNUSED(backend);
return ggml_kompute_supports_op(op); return ggml_vk_supports_op(op);
} }
static struct ggml_backend_i kompute_backend_i = { static struct ggml_backend_i kompute_backend_i = {