Function deepmd::get_env_nthreads

Contents

Function deepmd::get_env_nthreads#

Defined in File common.h

Function Documentation#

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

Get the number of threads from the environment variable.

A warning will be thrown if environment variables are not set.

Parameters:

num_intra_nthreads – [out] The number of intra threads. Read from DP_INTRA_OP_PARALLELISM_THREADS.
num_inter_nthreads – [out] The number of inter threads. Read from DP_INTER_OP_PARALLELISM_THREADS.