(fix) resolve the compiling error under the offline mode

Author: tsingmicro-public-e

python/setup.py

@@ -361,12 +361,15 @@ def download_and_copy(name, src_func, dst_path, variable, version, url_func):
         print(f'{YELLOW}downloading and extracting {url} ... {NC}', file=sys.stderr, flush=True)
         file = tarfile.open(fileobj=open_url(url), mode="r|*")
         file.extractall(path=tmp_path)
-    os.makedirs(os.path.split(dst_path)[0], exist_ok=True)
-    print(f'copy {src_path} to {dst_path} ...', file=sys.stderr, flush=True)
-    if os.path.isdir(src_path):
-        shutil.copytree(src_path, dst_path, dirs_exist_ok=True)
-    else:
-        shutil.copy(src_path, dst_path)
+    # In offline build mode the cache may be absent (e.g. backend doesn't need the
+    # NVIDIA toolkits), skip the copy in that case instead of failing.
+    if os.path.exists(src_path):
+        os.makedirs(os.path.split(dst_path)[0], exist_ok=True)
+        print(f'copy {src_path} to {dst_path} ...', file=sys.stderr, flush=True)
+        if os.path.isdir(src_path):
+            shutil.copytree(src_path, dst_path, dirs_exist_ok=True)
+        else:
+            shutil.copy(src_path, dst_path)
 
 
 # ---- cmake extension ----

python/setup_tools/utils/tools.py

@@ -132,7 +132,7 @@ def __init__(self):
         NetConfig.headers = {'User-Agent': NetConfig.user_agent}
 
     def download(self, url=None, path=None, file_name=None, mode=None, module=None, required=False):
-        if self.module_offline_handler.is_offline_build():
+        if self.module_offline_handler.is_offline_build() and self.module_offline_handler.offline_build_dir is not None:
             self.offline_copy(module, required)
             return
 
@@ -285,6 +285,9 @@ def handle_flagtree_hook(self, kargs):
 
     def handle_triton_origin_toolkits(self):
 
+        if self.offline_build_dir is None:
+            return
+
         # detect system/arch/version, the same with setup.py
         system = platform.system()
         arch = platform.machine()