frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] Update

Browse Source
This commit is contained in:
Richard Zou
2021-04-27 06:52:10 -07:00
committed by Jon Janzen
parent b7096ab83a
commit 70925e47c7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
functorch/functorch/csrc/DynamicLayer.cpp
View File

@ -224,9 +224,9 @@ static void sanityCheckStack(const c10::OperatorHandle& op, torch::jit::Stack* s
void dynamicLayerFrontFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) {
auto& dynamicLayerStack = dynamicLayerStackAccessor();
if (c10::show_dispatch_trace_enabled()) {
std::cout << "DLS size: " << dynamicLayerStack.size() << std::endl;
}
// if (c10::show_dispatch_trace_enabled()) {
// std::cout << "DLS size: " << dynamicLayerStack.size() << std::endl;
// }
if (dynamicLayerStack.size() == 0) {
sanityCheckStack(op, stack);
c10::impl::ExcludeDispatchKeyGuard guard(all_dynlayer_keyset);