目录
torch.get_num_interop_threads() → int
torch.set_num_interop_threads(int)
torch.get_num_threads() → int
Returns the number of threads used for parallelizing CPU operations
torch.set_num_threads(int)
Sets the number of threads used for parallelizing CPU operations. WARNING: To ensure that the correct number of threads is used, set_num_threads must be called before running eager, JIT or autograd code.
torch.get_num_interop_threads() → int
Returns the number of threads used for inter-op parallelism on CPU (e.g. in JIT interpreter)
torch.set_num_interop_threads(int)
Sets the number of threads used for interop parallelism (e.g. in JIT interpreter) on CPU. WARNING: Can only be called once and before any inter-op parallel work is started (e.g. JIT execution).
