saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "Disable DLL test"

Browse source 
This reverts commit e901cb93aa.
This commit is contained in:
Ryan Hill 2021-05-03 16:31:12 -07:00
parent acba6779df
commit 58a675ec2b
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
onnxruntime/test/python/onnxruntime_test_python.py
View file

@ -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: