[Performance] Fix Cython not being imported after switching to RTLD_L…

…OCAL (dmlc#2608) * fix * ? Co-authored-by: Jinjing Zhou <[email protected]>
feixian15 · Feb 2, 2021 · 47c9380 · 47c9380
1 parent 20d4977
commit 47c9380
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -90,12 +90,8 @@ def config_cython():
             subdir = "_cy2"
         ret = []
         path = "dgl/_ffi/_cython"
-        if os.name == 'nt':
-            library_dirs = ['dgl', '../build/Release', '../build']
-            libraries = ['libtvm']
-        else:
-            library_dirs = None
-            libraries = None
+        library_dirs = ['dgl', '../build/Release', '../build']
+        libraries = ['dgl']
         for fn in os.listdir(path):
             if not fn.endswith(".pyx"):
                 continue