fix unittest

pangyoki · Oct 27, 2021 · f4ce3d1 · f4ce3d1
1 parent 55cab10
commit f4ce3d1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_cuda_cudnn_version.py b/python/paddle/fluid/tests/unittests/test_cuda_cudnn_version.py
@@ -17,9 +17,10 @@
 
 
 class TestCPUVersion(unittest.TestCase):
-    if not paddle.is_compiled_with_cuda():
-        self.assertEqual(paddle.version.cuda(), 'False')
-        self.assertEqual(paddle.version.cudnn(), 'False')
+    def test_cuda_cudnn_version_in_cpu_package(self):
+        if not paddle.is_compiled_with_cuda():
+            self.assertEqual(paddle.version.cuda(), 'False')
+            self.assertEqual(paddle.version.cudnn(), 'False')
 
 
 if __name__ == '__main__':