aten

tongcu · Nov 21, 2017 · d9d7b1f · d9d7b1f
1 parent 8425bd8
commit d9d7b1f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/build.py b/build.py
@@ -25,13 +25,11 @@
 # 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']