set_multithreading_enabled

class torch.autograd.set_multithreading_enabled(mode)

Context-manager that sets multithreaded backwards on or off.

set_multithreading_enabled will enable or disable multithreaded backwards based on its argument mode. It can be used as a context-manager or as a function.

This context manager is thread local; it will not affect computation in other threads.

Parameters:

mode (bool) – Flag whether to enable multithreaded backwards (True), or disable (False).

Note

This API does not apply to forward-mode AD.