experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make MemoryUsageTracker::EndInstruction() return error status instead of crashing due to CHECK fails. This crashes the layout autotuner.

Browse Source 
PiperOrigin-RevId: 323507724
Change-Id: I5eafc0dbbb527164a9246131f1ad7f7d71ddcb44
This commit is contained in:
Mangpo Phothilimthana 2020-07-27 22:07:35 -07:00 committed by TensorFlower Gardener
parent 9efc52e47d
commit 8d35859243
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/hlo_rematerialization.cc
View File

@ -745,7 +745,7 @@ Status MemoryUsageTracker::EndInstruction() {
for (BufferId buffer_id : in_progress_item_->buffers_used) {
Buffer& buffer = buffers_.at(buffer_id);
buffer.unfinished_user_count--;
CHECK_GE(buffer.unfinished_user_count, 0)
TF_RET_CHECK(buffer.unfinished_user_count >= 0)
<< buffer.ToString() << " has negative unfinished user count.";
if (buffer.unfinished_user_count == 0) {
// Buffer is now dead.