Unverified Commit c960b119 authored by moto's avatar moto Committed by GitHub
Browse files

Split CUDA and CPU tests of torchscript_consistency (#644)

Currently there is no CUDA device for regular FB's infra, where `torchaudio`'s tests are ran.
Skipping these tests consistently raises a flag so we decided to split the tests into runnable and not-runnable on file-level.
parent b8187b4c
from common_utils import define_test_suites
from torchscript_consistency_impl import Functional, Transforms
define_test_suites(globals(), [Functional, Transforms], devices=['cpu'])
from common_utils import define_test_suites
from torchscript_consistency_impl import Functional, Transforms
define_test_suites(globals(), [Functional, Transforms], devices=['cuda'])
...@@ -608,6 +608,3 @@ class Transforms(common_utils.TestBaseMixin): ...@@ -608,6 +608,3 @@ class Transforms(common_utils.TestBaseMixin):
filepath = common_utils.get_asset_path("vad-go-mono-32000.wav") filepath = common_utils.get_asset_path("vad-go-mono-32000.wav")
waveform, sample_rate = torchaudio.load(filepath) waveform, sample_rate = torchaudio.load(filepath)
self._assert_consistency(T.Vad(sample_rate=sample_rate), waveform) self._assert_consistency(T.Vad(sample_rate=sample_rate), waveform)
common_utils.define_test_suites(globals(), [Functional, Transforms])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment