From ced04e1dff7f64279d068b9541591b4f5f01ca91 Mon Sep 17 00:00:00 2001 From: Ryan Voots Date: Thu, 26 Oct 2023 12:43:07 -0400 Subject: [PATCH] disable the error exit here, see if the pregen code works --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index e2c8df4..26256bf 100644 --- a/setup.py +++ b/setup.py @@ -99,9 +99,9 @@ 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]) 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 ret != 0: +# raise Exception(f"Failed generate with {ret}") +# sys.exit(-1) if not ROCM_VERSION: from distutils.sysconfig import get_python_lib