Re-insert gcc line closing #43

simonmeister · Aug 9, 2018 · e93f889 · e93f889
1 parent b98fafe
commit e93f889
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/e2eflow/ops.py b/src/e2eflow/ops.py
@@ -42,6 +42,9 @@ def compile(op=None):
                                 tf_inc, tf_lib)
         subprocess.check_output(nvcc_cmd, shell=True)
         gcc_cmd = "{} -std=c++11 -shared -o {} {} -fPIC -L " + cuda_dir + "/cuda/lib64 -lcudart {} -O2 -D GOOGLE_CUDA=1"
+        gcc_cmd = gcc_cmd.format(config['compile']['g++'],	
+                                 " ".join([fn_so, fn_cu_o, fn_cc]),	
+                                 tf_inc, tf_lib)
         subprocess.check_output(gcc_cmd, shell=True)