chore: bug fix in runtime (#3054)

pytorch · Aug 1, 2024 · 358c743 · 358c743
1 parent 23b4f1e
commit 358c743
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/runtime/execute_engine.cpp b/core/runtime/execute_engine.cpp
@@ -115,7 +115,7 @@ std::vector<at::Tensor> execute_engine(std::vector<at::Tensor> inputs, c10::intr
   bool need_cudagraphs_record = (CUDAGRAPHS_MODE && !_cudagraphs_validate_shapes(inputs, compiled_engine));
 
   // this is a buffer to store shape tensor input addresses throughout the runtime scope
-  std::list<std::vector<int32_t>> inputShapeTensorValues;
+  std::list<std::vector<int64_t>> inputShapeTensorValues;
 
   // Intialize outputs to be available throughout the succeeding scopes
   std::vector<at::Tensor> outputs(compiled_engine->num_io.second);