github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-12 12:47:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] skip torchaudio tracing test (#3211)

Browse Source 
* [hotfix] skip torchaudio tracing test

* fix lazy init test issue
This commit is contained in:
YuliangLiu0306
2023-03-24 12:15:33 +08:00
committed by GitHub
parent 78fd31f9c1
commit 045afa3ea2
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_fx/test_tracer/test_torchaudio_model/test_torchaudio_model.py
View File

@@ -6,7 +6,9 @@ from torchaudio_utils import trace_and_compare
from tests.kit.model_zoo import model_zoo
@pytest.mark.skipif(version.parse(torch.__version__) < version.parse('1.12.0'), reason='torch version < 12')
# We cannot handle the tensors constructed with constant during forward, such as ``torch.empty(0).to(device=Proxy.device)``
# TODO: We could handle this case by hijacking torch.Tensor.to function.
@pytest.mark.skip
def test_torchaudio_models():
torch.backends.cudnn.deterministic = True