Merge pull request #92 from dariooo512/master

Adding catch case for handling FileNotFound exception when using dll from .NET Core
Sergio0694 · Nov 1, 2019 · 1fe3211 · 1fe3211
2 parents e1e7e34 + 6299d70
commit 1fe3211
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/NeuralNetwork.NET/cuDNN/CuDnnService.cs b/NeuralNetwork.NET/cuDNN/CuDnnService.cs
@@ -1,4 +1,5 @@
 using System;
+using System.IO;
 using System.Linq;
 using System.Threading;
 using Alea;
@@ -75,7 +76,7 @@ public static bool IsAvailable
                     // Calling this directly could cause a crash in the <Module> loader due to the missing .dll files
                     return CuDnnSupportHelper.IsGpuAccelerationSupported();
                 }
-                catch (TypeInitializationException)
+                catch (Exception e) when (e is FileNotFoundException || e is TypeInitializationException)
                 {
                     // Missing .dll file
                     return false;