Function deepmd::get_env_nthreads

Function Documentation

void deepmd::get_env_nthreads(int &num_intra_nthreads, int &num_inter_nthreads)

Get the number of threads from the environment variable.

Parameters
  • num_intra_nthreads[out] The number of intra threads. Read from TF_INTRA_OP_PARALLELISM_THREADS.

  • num_inter_nthreads[out] The number of inter threads. Read from TF_INTER_OP_PARALLELISM_THREADS.