Address comments

pytorch · Apr 5, 2023 · 17c12b9 · 17c12b9
1 parent eb2377d
commit 17c12b9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/torch/csrc/jit/passes/peephole.cpp b/torch/csrc/jit/passes/peephole.cpp
@@ -276,7 +276,11 @@ struct PeepholeOptimizeImpl {
         if (string_type) {
           WithInsertPoint guard(node);
           std::string type_str = node->inputs().at(0)->node()->s(attr::value);
-          auto index = toIValue(node->inputs().at(1))->toInt();
+          auto maybe_index = toIValue(node->inputs().at(1));
+          int64_t index = 0;
+          if (maybe_index) {
+            index = maybe_index->toInt();
+          }
           auto device = c10::Device(type_str + ":" + std::to_string(index));
           auto output = node->owningGraph()->insertConstant(device);
           GRAPH_UPDATE(