cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

Update depthwise_conv2d_implicit_gemm.py

Open DingXiaoH opened this issue 2 years ago • 1 comments

Sometimes when the tensor format changes after this conv (e.g., NCHW -> NHWC for layer normalization), calling backward will raise an "input must be contiguous" error. Making the grad contiguous in advance simply solves it.

DingXiaoH avatar Apr 29 '22 17:04 DingXiaoH

CLA assistant check
All committers have signed the CLA.

CLAassistant avatar Apr 29 '22 17:04 CLAassistant