simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

Make this fail early when the actual problem happens

Browse source
This commit is contained in:
Ryan Voots 2023-10-26 09:38:59 -04:00
parent 9fb99f61e7
commit 153c085a32
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
setup.py
View file

@ -98,7 +98,10 @@ if BUILD_CUDA_EXT:
if platform.system() != 'Windows':
p = int(subprocess.run("cat /proc/cpuinfo | grep cores | head -1", shell=True, check=True, text=True, stdout=subprocess.PIPE).stdout.split(" ")[2])
subprocess.call(["python", "./autogptq_extension/qigen/generate.py", "--module", "--search", "--p", str(p)])
ret = subprocess.call(["python", "./autogptq_extension/qigen/generate.py", "--module", "--search", "--p", str(p)])
if ret != 0:
raise Exception(f"Failed generate with {ret}")
sys.exit(-1)
if not ROCM_VERSION:
from distutils.sysconfig import get_python_lib