whybeyoung
/
chatglm-6b-int4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bugs when compiling cpu kernels

This commit is contained in:
songxxzp 2023-04-06 20:09:56 +08:00
parent 1f34060390
commit 68873da921
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
quantization.py
View File

@ -144,6 +144,7 @@ class CPUKernel:
kernel_file = source_code[:-2] + ".so"
compile_command = "gcc -O3 -fPIC -std=c99 {} -shared -o {}".format(source_code, kernel_file)
print("Compiling", compile_command)
exit_state = os.system(compile_command)
else:
compile_command = "gcc -O3 -fPIC -std=c99 {} -shared -o {}".format(source_code, kernel_file)
print("Compiling", compile_command)