"git@developer.sourcefind.cn:one/TransferBench.git" did not exist on "291c49b8c1da8b8bcc6e19514824cccaf1475eb9"
Commit fa59855f authored by Zhaoheng Ni's avatar Zhaoheng Ni Committed by Facebook GitHub Bot
Browse files

Fix cuda test failure (#3363)

Summary:
Fix https://github.com/pytorch/audio/issues/3361

When adding FunctionalCUDAOnlyTest, the class should inherit from `TestBaseMixin` instead of `Functional`

Pull Request resolved: https://github.com/pytorch/audio/pull/3363

Reviewed By: atalman, osalpekar

Differential Revision: D46112084

Pulled By: nateanl

fbshipit-source-id: 67c6472fda98cb718e0fc53ab248beda745feab5
parent d850ff61
...@@ -1234,7 +1234,7 @@ class FunctionalCPUOnly(TestBaseMixin): ...@@ -1234,7 +1234,7 @@ class FunctionalCPUOnly(TestBaseMixin):
assert len(w) == 1 assert len(w) == 1
class FunctionalCUDAOnly(Functional): class FunctionalCUDAOnly(TestBaseMixin):
@nested_params( @nested_params(
[torch.half, torch.float, torch.double], [torch.half, torch.float, torch.double],
[torch.int32, torch.int64], [torch.int32, torch.int64],
......
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