Add assertion that 'pos' is in-bounds (#3466)

sungin-h · Nov 3, 2017 · 42de0df · 42de0df
1 parent a8efd88
commit 42de0df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/csrc/autograd/input_buffer.cpp b/torch/csrc/autograd/input_buffer.cpp
@@ -1,5 +1,6 @@
 #include "torch/csrc/autograd/input_buffer.h"
 
+#include "torch/csrc/assertions.h"
 #include "torch/csrc/autograd/functions/basic_ops.h"
 #include "torch/csrc/utils/auto_gpu.h"
 
@@ -10,6 +11,7 @@ InputBuffer::InputBuffer(size_t size)
   {}
 
 void InputBuffer::add(size_t pos, Variable var) {
+  TORCH_ASSERT(pos >= 0 && pos < buffer.size());
   if (!var.defined()) {
     return;
   }