Home
last modified time | relevance | path

Searched refs:num_threads_in_backwards (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp79 std::atomic<int32_t> num_threads_in_backwards; variable
82 num_threads_in_backwards++; in CompiledAutogradThreadingDebugCheck()
89 num_threads_in_backwards--; in release()
1405 num_threads_in_backwards.load() == 0 && prior != COMPILED_AUTOGRAD_POISON, in set_compiled_autograd()