Skip to content

GitLab

Commit 01946d40 authored by Hang Zhang's avatar Hang Zhang
Browse files

Merge branch 'master' of https://github.com/zhanghang1989/PyTorch-Encoding

parents fd2c8c76 d9d7b1f0
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
build.py
View file @ 01946d40
...@@ -25,13 +25,11 @@ subprocess.check_call(clean_cmd) ...@@ -25,13 +25,11 @@ subprocess.check_call(clean_cmd)
# build CUDA library # build CUDA library
os.environ['TORCH_BUILD_DIR'] = lib_path os.environ['TORCH_BUILD_DIR'] = lib_path
if platform.system() == 'Darwin': if platform.system() == 'Darwin':
os.environ['TH_LIBRARIES'] = os.path.join(lib_path,'libTH.1.dylib') os.environ['TH_LIBRARIES'] = os.path.join(lib_path,'libATen.1.dylib')
os.environ['THC_LIBRARIES'] = os.path.join(lib_path,'libTHC.1.dylib')
ENCODING_LIB = os.path.join(cwd, 'encoding/lib/libENCODING.dylib') ENCODING_LIB = os.path.join(cwd, 'encoding/lib/libENCODING.dylib')
else: else:
os.environ['TH_LIBRARIES'] = os.path.join(lib_path,'libTH.so.1') os.environ['TH_LIBRARIES'] = os.path.join(lib_path,'libATen.so.1')
os.environ['THC_LIBRARIES'] = os.path.join(lib_path,'libTHC.so.1')
ENCODING_LIB = os.path.join(cwd, 'encoding/lib/libENCODING.so') ENCODING_LIB = os.path.join(cwd, 'encoding/lib/libENCODING.so')
build_all_cmd = ['bash', 'encoding/make.sh'] build_all_cmd = ['bash', 'encoding/make.sh']
......
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