Code

A specialized Winograd Conv2d op#971

Code

Draft

bssrdf wants to merge 15 commits intoggerganov:masterggerganov/ggml:masterfrom bssrdf:add-winograd-conv2dbssrdf/ggml:add-winograd-conv2dCopy head branch name to clipboard

+1,417-3

Commits

Commits on Sep 26, 2024

• added two winograd ops

  

  bssrdf

  committed
  5b4e448

Commits on Sep 27, 2024

• added more checking for precondition to use winograd; switch to im2co if not satisfied

  

  bssrdf

  committed
  68c251b

• added source code for winograd kernel

  

  bssrdf

  committed
  2ccc67d

• winograd build ok

  

  bssrdf

  committed
  02a3cb1

• test now passed; for some reason, ggml_conv_2d didn't output correct results

  

  bssrdf

  committed
  3d80466

• remove debugging printouts

  

  bssrdf

  committed
  893ca79

Commits on Sep 28, 2024

• added a FP16 FX kernel to deal with fp16 filter data; no need to use FP32 buffer

  

  bssrdf

  committed
  6afbf6e

Commits on Sep 29, 2024

• sync from sd.cpp

  

  bssrdf

  committed
  0491858

• restore test-conv2d.cpp test

  

  bssrdf

  committed
  93c3da7

• restore src/CMakeLists.txt

  

  bssrdf

  committed
  4e8e0d4

• change mask to unsigned int; add __restric__ to various pointers

  

  bssrdf

  committed
  e0e94c4

• fix indentation

  

  bssrdf

  committed
  c8700ca

Commits on Oct 1, 2024

• skip if already computed in a preprocess step

  

  bssrdf

  committed
  c5d43a2

• added winograd conv2d to backend op tests

  

  bssrdf

  committed
  00ad37e

Commits on Oct 2, 2024

• add conv2d as a test case in test-backend-op

  

  bssrdf

  committed
  4f93d67