[BUG] PT: in customized OPs, ensure input tensors have continuous memory

[njzjz]

Our kernels assume the memory of input tensors is continuous. It seems the torch's autograd may return tensors with non-continuous memory. Need to call the contiguous method to ensure the memory is contiguous.

[njzjz]

xref: pytorch/pytorch#64529 (comment)