Rate this Page

torch.set_num_interop_threads#

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).