Merge pull request #6305 from notoraptor/fix-ctc-runtime-execution

theano/gpuarray/ctc.py

@@ -13,7 +13,7 @@ from theano.tensor.opt import register_canonicalize
 from theano.tensor.nnet.ctc import ctc_available
 
 import os
-import os.path
+import sys
 from . import pygpu
 
 
@@ -54,6 +54,14 @@ class GpuConnectionistTemporalClassification(gof.COp):
             lib_dirs += [ctc_available.path]
         return lib_dirs
 
+    def c_compile_args(self):
+        if ctc_available.path is not None:
+            if sys.platform != 'darwin' and ' ' in ctc_available.path:
+                return ['-Wl,-rpath,"' + ctc_available.path + '"']
+            else:
+                return ['-Wl,-rpath,' + ctc_available.path]
+        return []
+
     def c_libraries(self):
         return ["warpctc", "gpuarray"]
 

theano/tensor/nnet/ctc.py

 from __future__ import (division, absolute_import, print_function)
 import os
+import sys
 import theano.tensor as T
 from theano import config
 from theano import gof
@@ -126,6 +127,14 @@ class ConnectionistTemporalClassification(gof.COp, gof.OpenMPOp):
             lib_dirs += [ctc_available.path]
         return lib_dirs
 
+    def c_compile_args(self):
+        if ctc_available.path is not None:
+            if sys.platform != 'darwin' and ' ' in ctc_available.path:
+                return ['-Wl,-rpath,"' + ctc_available.path + '"']
+            else:
+                return ['-Wl,-rpath,' + ctc_available.path]
+        return []
+
     def c_libraries(self):
         return ["warpctc"]