Merge 5e22bad into 588cad5

chainer · Jul 16, 2019 · 1148162 · 1148162
2 parents 588cad5 + 5e22bad
commit 1148162
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/chainer/functions/noise/dropout.py b/chainer/functions/noise/dropout.py
@@ -44,7 +44,7 @@ def forward_cpu(self, x):
         return y,
 
     def forward_gpu(self, x):
-        if (chainer.should_use_cudnn('==always', 5000)
+        if (chainer.should_use_cudnn('>=auto', 5000)
                 and x[0].flags.c_contiguous
                 and self.mask is None
                 and not self.return_mask):

diff --git a/tests/chainer_tests/functions_tests/noise_tests/test_dropout.py b/tests/chainer_tests/functions_tests/noise_tests/test_dropout.py
@@ -8,7 +8,6 @@
 from chainer import gradient_check
 from chainer import testing
 from chainer.testing import attr
-from chainer.testing import backend
 
 
 @testing.parameterize(
@@ -17,7 +16,7 @@
     {'dtype': numpy.float64, 'ratio': 0.5},
     {'dtype': numpy.float64, 'ratio': 0.0},
 )
-@backend.inject_backend_tests(
+@testing.inject_backend_tests(
     ['test_forward', 'test_backward', 'test_double_backward',
      'test_immutable'],
     # CPU tests