#ifdef GPU_ENABLED
if (USE_GPU/* && (tmp.rows >= 512 || tmp.cols >= 512)*/)
{
- cpx_mtx_knk(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
+ cpx_mtx_knk_metal(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
}
else
{
cpx_mtx_knk(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
}
#else
- cpx_mtx_knk
- (
- tmp.ptr, filter.ptr, gate.ptr,
- tmp.rows, tmp.cols,
- filter.rows, filter.cols,
- gate.rows, gate.cols
- );
+ cpx_mtx_knk(tmp.ptr, filter.ptr, gate.ptr, filter.rows, filter.cols, gate.rows, gate.cols);
#endif