commit 670e66b84179ed8ecb196a09fefd3eea8eca60fd
author Ben Vanik <ben.vanik@gmail.com> Thu Apr 21 11:32:22 2022 -0700
committer GitHub <noreply@github.com> Thu Apr 21 11:32:22 2022 -0700
tree d03611373802a1118e320d5db33ef88a88e8a351
parent 6c6ac084644ea31670d1e80811cfbc8bc7b9c7f8

Printing the op when an allocation mapping fails. (#8952)

iree/compiler/Dialect/Stream/Transforms/ScheduleAllocation.cpp

diff --git a/iree/compiler/Dialect/Stream/Transforms/ScheduleAllocation.cpp b/iree/compiler/Dialect/Stream/Transforms/ScheduleAllocation.cpp
index 48687a3..8866f4d 100644
--- a/iree/compiler/Dialect/Stream/Transforms/ScheduleAllocation.cpp
+++ b/iree/compiler/Dialect/Stream/Transforms/ScheduleAllocation.cpp
@@ -346,7 +346,9 @@
         AsmState asmState(rootOp->getParentOp());
         llvm::dbgs() << "!! storage not pre-allocated for resource ";
         resource.printAsOperand(llvm::dbgs(), asmState);
-        llvm::dbgs() << ":\n";
+        llvm::dbgs() << "\n";
+        resource.getDefiningOp()->print(llvm::dbgs(), asmState);
+        llvm::dbgs() << "\ncurrent mappings:\n";
         for (auto mapping : resourceRangeMap) {
           llvm::dbgs() << "  * mapping ";
           mapping.first.printAsOperand(llvm::dbgs(), asmState);