getDispatchKeySetUnboxed Class — pytorch Architecture

Source Code

aten/src/ATen/core/dispatch/DispatchKeyExtractor.h lines 185–199

  template <class... Args>
  DispatchKeySet getDispatchKeySetUnboxed(const Args&... args) const {
    auto ks = detail::multi_dispatch_key_set(args...);
    // Keys that are fallthrough should be skipped
    if (requiresBitsetPerBackend_) {
      c10::impl::LocalDispatchKeySet tls =
          c10::impl::tls_local_dispatch_key_set();
      auto backend_idx =
          ((ks | tls.included_) - tls.excluded_).getBackendIndex();
      return impl::computeDispatchKeySet(
          ks, nonFallthroughKeysPerBackend_[backend_idx]);
    } else {
      return impl::computeDispatchKeySet(ks, nonFallthroughKeys_);
    }
  }