search

ACEsuit / ACEfit.jl

Generic Codes for Fitting ACE models
MIT License
7 stars 6 forks source link

Make python solvers print BLAS thread usage #35

Closed wcwitt closed 1 year ago

wcwitt commented 1 year ago

... using threadpool_info()

wcwitt commented 1 year ago

I've verified it is possible to inspect and extract the BLAS threads using, for example,

    controller = pyimport("threadpoolctl")["ThreadpoolController"]()
    controller.limit(limits=nprocs, user_api="blas")
    pyimport("pprint")["pprint"](controller.select(user_api="blas").info())

For now, I decided to do it only for ace_fit.jl, not at the solver level. Might return to this though.