[BUG] Tests are failing with cuda

[bruAristimunha]

Hey! Thanks for the review.

The failing tests when running pytest locally are:

• FAILED tests/test_integration.py::test_module_dtype[WaveletConv-float32-cuda] - RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

• FAILED tests/test_integration.py::test_module_dtype[WaveletConv-float64-cuda] - RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument weight in method wrapper_CUDA___slow_conv2d_forward)

• FAILED tests/test_integration.py::test_module_dtype[WaveletConv-complex64-cuda] - RuntimeError: Input type (CPUComplexFloatType) and weight type (CUDAComplexFloatType) should be the same

• FAILED tests/test_integration.py::test_module_dtype[WaveletConv-complex128-cuda] - RuntimeError: Input type (CPUComplexDoubleType) and weight type (CUDAComplexDoubleType) should be the same

• FAILED tests/test_modules.py::test_bimap_instantiation_and_shape[Kaiming (Parametrized)] - AssertionError: Failed device check for Kaiming (Parametrized)

• FAILED tests/test_modules.py::test_bimap_instantiation_and_shape[Orthogonal (Parametrized)] - AssertionError: Failed device check for Orthogonal (Parametrized)

• FAILED tests/test_modules.py::test_bimap_instantiation_and_shape[Stiefel (Non-Param)] - AssertionError: Failed device check for Stiefel (Non-Param)

• FAILED tests/test_modules.py::test_bimap_instantiation_and_shape[Dim Increase Kaiming (Param)] - AssertionError: Failed device check for Dim Increase Kaiming (Param)

Most of them seem related to CUDA/device mismatch issues.

Originally posted by @Emma5938 in #26 (comment)