Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Increase CPU usage cap from 400% to higher number, better if configurable #62

Open
tuobulatuo opened this issue Sep 25, 2023 · 1 comment

Comments

@tuobulatuo
Copy link

Hi Song Lab Team,

I noticed that the CPU usage is capped at 4 threads (400%). Can you please make it a configurable number? For example, if I have a machine with 10 CPUs, I can use all of them.

Thanks!
Alex

@RaymondWang0
Copy link
Collaborator

Hi @tuobulatuo, thanks for your valuable suggestion. We've made a quick fix to support a configurable number of threads for matrix multiplication on CPU platforms, as shown in README. You can now put this in the command line to configure different numbers of threads:

./chat <model_name> <precision> <num_threads>

I hope this helps. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants