RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input.

[hezhu1996]

Hi @ResearchingDexter , there might be something wrong with the

a_b=list(a_b.chunk(self.M,dim=1))
And it always give me this error unless I turn of cudnn which makes it very slow:
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input.

Any suggestion?

[ResearchingDexter]

I guess that the cuDNN error about : CUDNN_STATUS_NOT_SUPPORTED was caused by the list of the a_b=list(a_b.chunk(self.M,dim=1)) , which not supported to make cuda tensor convert to the form like list[cudaTensor]. I supposed that used the form like a, b=a_b.chunk(self.M, dim=1) when M==2 to resolved the error. And thanks for your Issue.