/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
src/backends/cpu/logic.hpp
63 строки
2 KB
kolkir
Refactor tensor logic simd ops with map
16 мар 2025, 23:44
16 мар 2025, 23:44
8996bad
Код
Авторство
О чём код?
// should be included in only ones section after arithmetics-inl.hpp #pragma once #include <hwy/highway.h> #include "logic-inl.hpp" #include <adept/types.hpp> #if HWY_ONCE namespace adept::detail { // callers for SIMD function overloads struct gt { template <typename DataType> static void apply(const DataType* x_array, float32_t scalar, DataType* z_array, index_t size, bool aligned) { HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(gt_vector_scalar<DataType>) (x_array, scalar, z_array, size, aligned); } }; struct ge { template <typename DataType> static void apply(const DataType* x_array, float32_t scalar, DataType* z_array, index_t size, bool aligned) { HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(ge_vector_scalar<DataType>) (x_array, scalar, z_array, size, aligned); } }; struct lt { template <typename DataType> static void apply(const DataType* x_array, float32_t scalar, DataType* z_array, index_t size, bool aligned) { HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(lt_vector_scalar<DataType>) (x_array, scalar, z_array, size, aligned); } }; struct le { template <typename DataType> static void apply(const DataType* x_array, float32_t scalar, DataType* z_array, index_t size, bool aligned) { HWY_EXPORT_AND_DYNAMIC_DISPATCH_T(le_vector_scalar<DataType>) (x_array, scalar, z_array, size, aligned); } }; } // namespace adept::detail #endif // HWY_ONCE