-
Notifications
You must be signed in to change notification settings - Fork 93
Add a preference option for DFTK threading #972
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
Conversation
Bumps [julia-actions/cache](https://github.com/julia-actions/cache) from 1 to 2. - [Release notes](https://github.com/julia-actions/cache/releases) - [Commits](julia-actions/cache@v1...v2) --- updated-dependencies: - dependency-name: julia-actions/cache dependency-type: direct:production update-type: version-update:semver-major ... Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
This works fine locally, so let's merge? |
Does this allow to enable and disable it without launching again DFTK? For some computations, it makes sense to have multiple/single thread DFTK in the same script. |
Nope, I figured let's do the simple thing and require a reboot... |
Thanks! @mfherbst OK to merge? |
src/common/threading.jl
Outdated
FFTW.set_num_threads(n_fft) | ||
BLAS.set_num_threads(n_blas) | ||
mpi_master() && @info "Threading setup:" n_fft n_blas n_julia | ||
mpi_master() && @info "Threading setup: $(Threads.nthreads()) Julia threads, $n_DFTK DFTK threads, $n_fft FFT threads, $n_blas BLAS threads" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why? The other option gave a nice table?
Also line too long :-P
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ah I didn't know this, pretty nice
Review done |
Draft because untested so there's probably typos, I'll put it through its paces sometime later, but should be complete.