add stage to log (#9249)

apache · Oct 12, 2021 · 8725eb5 · 8725eb5
1 parent 01744d1
commit 8725eb5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/te/operation/tensorize.cc b/src/te/operation/tensorize.cc
@@ -337,7 +337,8 @@ void VerifyTensorizeBody(const ComputeOpNode* self, const Stage& stage,
     }
     ICHECK(expr_equal(lhs, rhs)) << "Failed to match the compute with TensorIntrin " << intrin->name
                                  << "'s declaration "
-                                 << " provided= " << lhs << ", intrin=  " << rhs;
+                                 << " provided= " << lhs << ", intrin=  " << rhs
+                                 << ", running this stage: " << stage;
   }
 }