Skip building native architecture when specifying arch list (#4727)

2026-04-23 00:17:25 +08:00 · 2025-10-31 20:32:46 +08:00
parent 3cbca75cc8
commit 27746026c1
1 changed files with 7 additions and 6 deletions
@@ -131,12 +131,13 @@ def get_sm_version(archs):
    Get sm version of paddle.
    """
    arch_set = set(archs)
-    try:
-        prop = paddle.device.cuda.get_device_properties()
-        cc = prop.major * 10 + prop.minor
-        arch_set.add(cc)
-    except ValueError:
-        pass
+    if len(arch_set) == 0:
+        try:
+            prop = paddle.device.cuda.get_device_properties()
+            cc = prop.major * 10 + prop.minor
+            arch_set.add(cc)
+        except ValueError:
+            pass
    return list(arch_set)