remove build dir in setup.py

Signed-off-by: Charlene Yang <[email protected]>

Author: cyanguwa

setup.py

@@ -377,7 +377,6 @@ def run(self) -> None:
             if isinstance(ext, CMakeExtension):
                 print(f"Building CMake extension {ext.name}")
                 with tempfile.TemporaryDirectory() as build_dir:
-                    build_dir = '/code/pr-graph-api/v5/TransformerEngine/build'
                     build_dir = Path(build_dir)
                     package_path = Path(self.get_ext_fullpath(ext.name))
                     install_dir = package_path.resolve().parent
@@ -497,8 +496,8 @@ def setup_pytorch_extension() -> setuptools.Extension:
     else:
         if version >= (11, 2):
             nvcc_flags.extend(["--threads", "4"])
-#        if version >= (11, 0):
-#            nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
+        if version >= (11, 0):
+            nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
         if version >= (11, 8):
             nvcc_flags.extend(["-gencode", "arch=compute_90,code=sm_90"])
 
@@ -548,8 +547,8 @@ def setup_paddle_extension() -> setuptools.Extension:
     cxx_flags = ["-O3"]
     nvcc_flags = [
         "-O3",
-#        "-gencode",
-#        "arch=compute_70,code=sm_70",
+        "-gencode",
+        "arch=compute_70,code=sm_70",
         "-U__CUDA_NO_HALF_OPERATORS__",
         "-U__CUDA_NO_HALF_CONVERSIONS__",
         "-U__CUDA_NO_BFLOAT16_OPERATORS__",
@@ -569,8 +568,8 @@ def setup_paddle_extension() -> setuptools.Extension:
     else:
         if version >= (11, 2):
             nvcc_flags.extend(["--threads", "4"])
-#        if version >= (11, 0):
-#            nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
+        if version >= (11, 0):
+            nvcc_flags.extend(["-gencode", "arch=compute_80,code=sm_80"])
         if version >= (11, 8):
             nvcc_flags.extend(["-gencode", "arch=compute_90,code=sm_90"])