Code

[CUTLASS] Conv2d dgrad#10110

Code

Merged

masahi merged 7 commits intoapache:mainapache/tvm:mainfrom masahi:cutlass-dgradmasahi/tvm:cutlass-dgradCopy head branch name to clipboardFeb 1, 2022

+474-115

Commits

Commits on Feb 1, 2022

• add conv2d transpose nhwc cudnn test

  masahi

  committed
  a49f7c2

  Copy full SHA for a49f7c2

• support conv2d transpose nhwc direct offload to cudnn

  masahi

  committed
  74503b0

  Copy full SHA for 74503b0

• add cutlass dgrad support

  masahi

  committed
  3e98120

  Copy full SHA for 3e98120

• remove unused arg

  masahi

  committed
  83e7af7

  Copy full SHA for 83e7af7

• allow target none

  masahi

  committed
  a031085

  Copy full SHA for a031085

• fix beta initiaization condition

  masahi

  committed
  2bb1705

  Copy full SHA for 2bb1705

• disable dynamic dense fp16 test since it fails on cuda 11.6

  masahi

  committed
  e0ca0e6

  Copy full SHA for e0ca0e6