TypeError: sru_forward(): incompatible function arguments.

[KP-Zhang]

Thank you for your work. When running the example, I encounter the following error.

Pytorch 1.6.0
sur: 2.5.1

File "C:\Anaconda3\lib\site-packages\sru\cuda_functional.py", line 83, in forward
is_custom
TypeError: sru_forward(): incompatible function arguments. The following argument types are supported:

tensor([[0., 0., 0., ..., 0., 0., 0.],
[0., 0., 0., ..., 0., 0., 0.],
[0., 0., 0., ..., 0., 0., 0.],
...,
[0., 0., 0., ..., 0., 0., 0.],
[0., 0., 0., ..., 0., 0., 0.],
[0., 0., 0., ..., 0., 0., 0.]], device='cuda:0'), tensor([]), tensor([], dtype=torch.uint8), 20, 32, 128, 3, 0, 1, False