TurboTransformers
TurboTransformers copied to clipboard
Set num threads ineffective
trafficstars
turbo_transformers.set_num_threads(?)
Whatever number I put in there, it is always using 4 threads.
In your command line set
export OMP_NUM_THREADS=?
Does not seem to work either ...
If you are using onnxrt as backend. The multiple-threading is not managed by OMP env. export MKL_VERBOSE=1. See the number of threads used for GEMM.