We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f929b0a commit e55b83eCopy full SHA for e55b83e
core/partitioning/shape_analysis.cpp
@@ -1,5 +1,6 @@
1
#include "core/partitioning/shape_analysis.h"
2
#include <ATen/ATen.h>
3
+#include <inttypes.h>
4
#include "core/util/prelude.h"
5
#include "torch/csrc/jit/api/module.h"
6
#include "torch/csrc/jit/passes/constant_pooling.h"
@@ -139,7 +140,7 @@ void getSegmentsOutputByRunning(
139
140
}
141
if (cur_ivalue.toTensor().sizes().size() == 0) {
142
// handle Scalar types, which has sizes of []
- 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}))));
144
} else {
145
input_shapes.push_back(util::toVec(util::toDims(cur_ivalue.toTensor().sizes())));
146
0 commit comments