search

EnzymeAD / Reactant.jl

MIT License
46 stars 2 forks source link

How to control the number of threads? #63

Open avik-pal opened 1 month ago

avik-pal commented 1 month ago

Setting up some benchmarking, is there a way to control the number of threads being used? I am guessing the julia threads count doesn't propagate.

avik-pal commented 1 month ago

Will still need to confirm if this works. But https://github.com/google/jax/issues/743 is probably the way to go.

mofeing commented 1 month ago

Great! I was questioning the same thing. Can't it be controlled through PJRT API?