/
vshmidt
/
llama.cpp
Обзор
Документация
Войти
/
vshmidt
/
llama.cpp
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
ggml/src/ggml-cpu/traits.cpp
36 строк
1 KB
Diego Devesa
ggml : fix fallback to CPU for ununsupported ops (#15118)
06 авг 2025, 15:37
Не верифицирован
06 авг 2025, 15:37
0d88315
Код
Авторство
О чём код?
#include "traits.h" #include "ggml-backend-impl.h" #include "ggml-backend.h" namespace ggml::cpu { tensor_traits::~tensor_traits() {} extra_buffer_type::~extra_buffer_type() {} } // namespace ggml::cpu bool ggml_cpu_extra_compute_forward(struct ggml_compute_params * params, struct ggml_tensor * op) { for (auto extra : ggml_backend_cpu_get_extra_buffer_types()) { if (extra && extra->context) { auto buf_extra = (ggml::cpu::extra_buffer_type *) extra->context; auto tensor_traits = buf_extra->get_tensor_traits(op); if (tensor_traits && tensor_traits->compute_forward(params, op)) { return true; } } } return false; } bool ggml_cpu_extra_work_size(int n_threads, const struct ggml_tensor * op, size_t * size) { for (auto extra : ggml_backend_cpu_get_extra_buffer_types()) { if (extra && extra->context) { auto buf_extra = (ggml::cpu::extra_buffer_type *) extra->context; auto tensor_traits = buf_extra->get_tensor_traits(op); if (tensor_traits && tensor_traits->work_size(n_threads, op, *size)) { return true; } } } return false; }