Преглед на файлове

Add more informative message for CUDA error

nagadomi преди 9 години
родител
ревизия
958dcfaeb8
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      lib/w2nn.lua

+ 1 - 1
lib/w2nn.lua

@@ -15,7 +15,7 @@ if w2nn then
 else
    local state, ret = pcall(load_cunn)
    if not state then
-      error("Failed to load CUDA modules. Please check the CUDA Settings.")
+      error("Failed to load CUDA modules. Please check the CUDA Settings.\n---\n" .. ret)
    end
    pcall(load_cudnn)
    w2nn = {}