Permalink
Browse files

Re-insert gcc line closing #43

  • Loading branch information...
simonmeister committed Aug 9, 2018
1 parent b98fafe commit e93f88988c50007cdb4538276a599a86941108d1
Showing with 3 additions and 0 deletions.
  1. +3 −0 src/e2eflow/ops.py
View
@@ -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)

0 comments on commit e93f889

Please sign in to comment.