update

pyg-team · Apr 6, 2024 · 0a6d0a9 · 0a6d0a9
1 parent af49e9b
commit 0a6d0a9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 # Environment flags to control different options
 #
-#   USE_MKL_BLAS=1
-#     enables use of MKL BLAS (requires PyTorch to be built with MKL support)
+# - USE_MKL_BLAS=1:
+#   Enables use of MKL BLAS (requires PyTorch to be built with MKL support)
 
 import importlib
 import os
@@ -44,6 +44,8 @@ def build_extension(self, ext):
         import torch
 
         extdir = osp.abspath(osp.dirname(self.get_ext_fullpath(ext.name)))
+        print(ext)
+        print(extdir)
         self.build_type = "DEBUG" if self.debug else "RELEASE"
         if self.debug is None:
             if CMakeBuild.check_env_flag("DEBUG"):