Skip to content

S
seminar-breakout
Commit 88d7baeb authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

use INCLUDE_DIR as an environment variable

parent dc575db9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/user_ops/__init__.py
View file @ 88d7baeb
...@@ -11,7 +11,7 @@ __all__ = ['zmq_recv'] ...@@ -11,7 +11,7 @@ __all__ = ['zmq_recv']
include_dir = tf.sysconfig.get_include() include_dir = tf.sysconfig.get_include()
file_dir = os.path.dirname(os.path.abspath(__file__)) file_dir = os.path.dirname(os.path.abspath(__file__))
compile_cmd = 'make INCLUDE_DIR:="-isystem {}" -C "{}"'.format(include_dir, file_dir) compile_cmd = 'INCLUDE_DIR="-isystem {}" make -C "{}"'.format(include_dir, file_dir)
print("Compiling user ops ...") print("Compiling user ops ...")
ret = os.system(compile_cmd) ret = os.system(compile_cmd)
if ret != 0: if ret != 0:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment