chore!: Bumping version numbers to 0.1.0

BREAKING CHANGE: Version is being bumped to version 0.1.0a0 to target PyTorch 1.6.0 Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Aug 25, 2020 · b84c90b · b84c90b
1 parent 50c7eda
commit b84c90b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/cpp/api/include/trtorch/macros.h b/cpp/api/include/trtorch/macros.h
@@ -20,8 +20,8 @@
 #define STR(x) XSTR(x)
 
 #define TRTORCH_MAJOR_VERSION 0
-#define TRTORCH_MINOR_VERSION 0
-#define TRTORCH_PATCH_VERSION 3
+#define TRTORCH_MINOR_VERSION 1
+#define TRTORCH_PATCH_VERSION 0
 #define TRTORCH_VERSION STR(TRTORCH_MAJOR_VERSION) \
                         "." STR(TRTORCH_MINOR_VERSION) \
                         "." STR(TRTORCH_PATCH_VERSION)
diff --git a/py/setup.py b/py/setup.py
@@ -16,7 +16,7 @@
 
 dir_path = os.path.dirname(os.path.realpath(__file__))
 
-__version__ = '0.0.3'
+__version__ = '0.1.0a0'
 
 CXX11_ABI = False
 
@@ -158,7 +158,8 @@ def run(self):
     cpp_extension.CUDAExtension('trtorch._C',
                                 ['trtorch/csrc/trtorch_py.cpp'],
                                 library_dirs=[
-                                    dir_path + '/trtorch/lib/'
+                                    (dir_path + '/trtorch/lib/'),
+                                    "/opt/conda/lib/python3.6/config-3.6m-x86_64-linux-gnu"
                                 ],
                                 libraries=[
                                     "trtorch"
@@ -176,7 +177,14 @@ def run(self):
                                     "-Wno-deprecated-declarations",
                                     "-Wl,--no-as-needed",
                                     "-ltrtorch",
-                                    "-Wl,-rpath,$ORIGIN/lib"
+                                    "-Wl,-rpath,$ORIGIN/lib",
+                                    "-lpthread",
+                                    "-ldl",
+                                    "-lutil",
+                                    "-lrt",
+                                    "-lm",
+                                    "-Xlinker",
+                                    "-export-dynamic"
                                 ] + (["-D_GLIBCXX_USE_CXX11_ABI=1"] if CXX11_ABI else ["-D_GLIBCXX_USE_CXX11_ABI=0"]),
                                 undef_macros=[ "NDEBUG" ]
                             )