Home / Class/ wrapPropagateTLSState Class — pytorch Architecture

wrapPropagateTLSState Class — pytorch Architecture

Architecture documentation for the wrapPropagateTLSState class in ThreadLocalState.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/ThreadLocalState.h lines 116–124

template <typename T>
auto wrapPropagateTLSState(T callback) {
  return [tls_state = ThreadLocalState(),
          callback = std::move(callback)](auto&&... args) {
    ThreadLocalStateGuard g(tls_state);
    // Propagate value returned by callback().
    return callback(std::forward<decltype(args)>(args)...);
  };
}

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free