提交 66f6e192 authored 作者: Vikram's avatar Vikram

.excluding('cudnn') removed for asymmetric and causal GPU tests

上级 5ae986b1
...@@ -276,11 +276,11 @@ class TestUnsharedGpuCorr2d(TestUnsharedConv): ...@@ -276,11 +276,11 @@ class TestUnsharedGpuCorr2d(TestUnsharedConv):
class TestAsymmetricGpu(TestAsymmetricPadding): class TestAsymmetricGpu(TestAsymmetricPadding):
mode = mode_with_gpu.excluding('cudnn') mode = mode_with_gpu
conv2d_op = GpuCorrMM conv2d_op = GpuCorrMM
conv2d_gradw_op = GpuCorrMM_gradWeights conv2d_gradw_op = GpuCorrMM_gradWeights
conv2d_gradi_op = GpuCorrMM_gradInputs conv2d_gradi_op = GpuCorrMM_gradInputs
class TestCausalGpuCorr(TestCausalConv): class TestCausalGpuCorr(TestCausalConv):
mode = mode_with_gpu.excluding('cudnn') mode = mode_with_gpu
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论