oneflow.set_num_threads

oneflow.set_num_threads()

Sets the number of threads used for intraop parallelism on CPU.

Warning

To ensure that the correct number of threads is used, set_num_threads must be called before running eager, eager globe or ddp.