Skip to content

GitLab

"...api/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "b3e9dfced7c9e8d00f646c710766b532383f04c6"
Commit d9d7b1f0 authored by Hang Zhang's avatar Hang Zhang
Browse files

aten

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