Skip to content

Commit e55b83e

Browse files
gcuendet-cognexgcuendet
authored andcommitted
Fix windows compilation error
Signed-off-by: Gabriel Cuendet <[email protected]>
1 parent f929b0a commit e55b83e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

core/partitioning/shape_analysis.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
#include "core/partitioning/shape_analysis.h"
22
#include <ATen/ATen.h>
3+
#include <inttypes.h>
34
#include "core/util/prelude.h"
45
#include "torch/csrc/jit/api/module.h"
56
#include "torch/csrc/jit/passes/constant_pooling.h"
@@ -139,7 +140,7 @@ void getSegmentsOutputByRunning(
139140
}
140141
if (cur_ivalue.toTensor().sizes().size() == 0) {
141142
// handle Scalar types, which has sizes of []
142-
input_shapes.push_back(util::toVec(util::toDims(c10::List<long int>({1}))));
143+
input_shapes.push_back(util::toVec(util::toDims(c10::List<int64_t>({1}))));
143144
} else {
144145
input_shapes.push_back(util::toVec(util::toDims(cur_ivalue.toTensor().sizes())));
145146
}

0 commit comments

Comments
 (0)