From e1b7f5507c386ba2e25aa04639c8a7f01837ed39 Mon Sep 17 00:00:00 2001 From: ming1753 Date: Fri, 31 Oct 2025 15:39:46 +0800 Subject: [PATCH] Skip building native architecture when specifying arch list --- custom_ops/setup_ops.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/custom_ops/setup_ops.py b/custom_ops/setup_ops.py index c82dfe411ec..81f86b54300 100644 --- a/custom_ops/setup_ops.py +++ b/custom_ops/setup_ops.py @@ -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)