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