From 58a675ec2b0acb3b68f0e21547df91c2952d6eae Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Mon, 3 May 2021 16:31:12 -0700 Subject: [PATCH] Revert "Disable DLL test" This reverts commit e901cb93aafd537af088fa29629029a479926362. --- .../test/python/onnxruntime_test_python.py | 30 +++++++++---------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/onnxruntime/test/python/onnxruntime_test_python.py b/onnxruntime/test/python/onnxruntime_test_python.py index 1e790cae50..14d782573b 100644 --- a/onnxruntime/test/python/onnxruntime_test_python.py +++ b/onnxruntime/test/python/onnxruntime_test_python.py @@ -210,22 +210,22 @@ class TestInferenceSession(unittest.TestCase): option = {'invalid_option': 123} sess.set_providers(['CUDAExecutionProvider'], [option]) -# libnames = ('libcuda.so', 'libcuda.dylib', 'cuda.dll') -# for libname in libnames: -# try: -# cuda = ctypes.CDLL(libname) -# runBaseTest1() -# runBaseTest2() -# runAdvancedTest() + libnames = ('libcuda.so', 'libcuda.dylib', 'cuda.dll') + for libname in libnames: + try: + cuda = ctypes.CDLL(libname) + runBaseTest1() + runBaseTest2() + runAdvancedTest() -# except OSError: -# continue -# else: -# break -# else: -# runBaseTest1() -# runBaseTest2() -# # raise OSError("could not load any of: " + ' '.join(libnames)) + except OSError: + continue + else: + break + else: + runBaseTest1() + runBaseTest2() + # raise OSError("could not load any of: " + ' '.join(libnames)) def testInvalidSetProviders(self): with self.assertRaises(RuntimeError) as context: