def test_check_cudnn_version(self): with self.assertRaises(RuntimeError): build.get_cudnn_version() self.assertTrue(build.check_cudnn_version(self.compiler, self.settings)) self.assertIsInstance(build.get_cudnn_version(), int) self.assertIsInstance(build.get_cudnn_version(True), str)
def test_check_cudnn_version(self): self.assertTrue(build.check_cudnn_version( self.compiler, self.settings))