Skip to content

Commit 358c743

Browse files
authored
chore: bug fix in runtime (#3054)
1 parent 23b4f1e commit 358c743

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/runtime/execute_engine.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ std::vector<at::Tensor> execute_engine(std::vector<at::Tensor> inputs, c10::intr
115115
bool need_cudagraphs_record = (CUDAGRAPHS_MODE && !_cudagraphs_validate_shapes(inputs, compiled_engine));
116116

117117
// this is a buffer to store shape tensor input addresses throughout the runtime scope
118-
std::list<std::vector<int32_t>> inputShapeTensorValues;
118+
std::list<std::vector<int64_t>> inputShapeTensorValues;
119119

120120
// Intialize outputs to be available throughout the succeeding scopes
121121
std::vector<at::Tensor> outputs(compiled_engine->num_io.second);

0 commit comments

Comments
 (0)