import torch

from torchaudio_unittest.common_utils import (
    PytorchTestCase,
    skipIfNoCuda,
)
from . transforms_test_impl import TransformsTestBase


@skipIfNoCuda
class TransformsCUDAFloat32Test(TransformsTestBase, PytorchTestCase):
    device = 'cuda'
    dtype = torch.float32


@skipIfNoCuda
class TransformsCUDAFloat64Test(TransformsTestBase, PytorchTestCase):
    device = 'cuda'
    dtype = torch.float64
