def METHOD_NAME(device):
    from speechbrain.processing.features import DCT
    compute_dct = DCT(input_size=40)
    inputs = torch.randn([10, 101, 40], device=device)
    assert torch.jit.trace(compute_dct, inputs)