Code

[CUTLASS] Initial conv2d support#9595

Code

Merged

comaniac merged 24 commits intoapache:mainapache/tvm:mainfrom masahi:cutlass-conv2dmasahi/tvm:cutlass-conv2dCopy head branch name to clipboardDec 2, 2021

+776-25

Commits

Commits on Dec 1, 2021

• Add initial conv generator

  masahi

  committed
  2dc60db

  Copy full SHA for 2dc60db

• added conv2d pattern

  masahi

  committed
  b4b0ca4

  Copy full SHA for b4b0ca4

• profile by gemm profiler

  masahi

  committed
  aa94787

  Copy full SHA for aa94787

• remove conv2d profiler for now

  masahi

  committed
  75f671b

  Copy full SHA for 75f671b

• remove unused code

  masahi

  committed
  ceeecfc

  Copy full SHA for ceeecfc

• add default

  masahi

  committed
  ef7e305

  Copy full SHA for ef7e305

• minor fix, profiling working

  masahi

  committed
  f87971b

  Copy full SHA for f87971b

• start codegen

  masahi

  committed
  30de42b

  Copy full SHA for 30de42b

• generated code compiled

  masahi

  committed
  a7a2173

  Copy full SHA for a7a2173

• fixed layout initialization

  masahi

  committed
  60666bc

  Copy full SHA for 60666bc

• matched with autotvm tensorcore result

  masahi

  committed
  6ff06dc

  Copy full SHA for 6ff06dc

• test refactor

  masahi

  committed
  da610ba

  Copy full SHA for da610ba

• minor cleanup

  masahi

  committed
  1629dfd

  Copy full SHA for 1629dfd

• remove iteration algo "Analytic"

  masahi

  committed
  8566d4f

  Copy full SHA for 8566d4f

• add test for dynamic batch conv2d

  masahi

  committed
  556b13e

  Copy full SHA for 556b13e

• pass dl tensor as output too

  masahi

  committed
  ad97f67

  Copy full SHA for ad97f67

• support conv2d dynamic shape in codegen

  masahi

  committed
  1ab88d5

  Copy full SHA for 1ab88d5

• test working

  masahi

  committed
  cb3548f

  Copy full SHA for cb3548f

• lint

  masahi

  committed
  1959b05

  Copy full SHA for 1959b05

• simplify codegen

  masahi

  committed
  b8f1bfd

  Copy full SHA for b8f1bfd

• fix weird formatting

  masahi

  committed
  71a1af8

  Copy full SHA for 71a1af8

• typo fix

  masahi

  committed
  0dd9e75

  Copy full SHA for 0dd9e75

• check if cutlass is enabled in the test

  masahi

  committed
  67d6312

  Copy full SHA for 67d6312

• simplify gen_conv2d.py

  masahi

  committed
  bfe22bf

  Copy full SHA for bfe22bf