From 9a759e9fb81eb02f0af6d9ef2c55d3f401125a54 Mon Sep 17 00:00:00 2001 From: jzl Date: Fri, 25 Apr 2025 11:42:38 +0800 Subject: [PATCH] fix: make cpufeature a local import --- setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 67ab60f..c5bf128 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,6 @@ import torch import torch.version from wheel.bdist_wheel import bdist_wheel as _bdist_wheel from setuptools import setup, Extension -from cpufeature.extension import CPUFeature from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CUDA_HOME, ROCM_HOME try: from torch_musa.utils.simple_porting import SimplePorting @@ -188,6 +187,8 @@ class VersionInfo: raise ValueError( "Unsupported cpu Instructions: {}".format(flags_line)) elif sys.platform == "win32": + from cpufeature.extension import CPUFeature + if CPUFeature.get("AVX512bw", False): return 'fancy' if CPUFeature.get("AVX512f", False):