From 9a759e9fb81eb02f0af6d9ef2c55d3f401125a54 Mon Sep 17 00:00:00 2001 From: jzl Date: Fri, 25 Apr 2025 11:42:38 +0800 Subject: [PATCH 1/2] 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): From 82920e7943962295dab9344b0325cdbd94a78ae1 Mon Sep 17 00:00:00 2001 From: liam Date: Fri, 25 Apr 2025 06:49:56 +0000 Subject: [PATCH 2/2] :spakles: update requirements for cpufeature --- requirements-local_chat.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-local_chat.txt b/requirements-local_chat.txt index 855b360..6c01add 100644 --- a/requirements-local_chat.txt +++ b/requirements-local_chat.txt @@ -3,7 +3,7 @@ transformers==4.43.2 numpy torch>=2.3.0 packaging -cpufeature +cpufeature; sys_platform == 'win32' or sys_platform == 'Windows' protobuf tiktoken blobfile