saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix JSEP build break (#16636)

Browse source 
### Description
fix JSEP build break. the build break was caused by enabling
`-Wshorten-64-to-32` while merging the CI.
This commit is contained in:
Yulong Wang 2023-07-09 08:53:11 -07:00 committed by GitHub
parent 00e8f2a2a9
commit 67f4cd54fa
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/providers/js/operators/conv_transpose.h
View file

@ -63,7 +63,7 @@ class ConvTranspose : public JsKernel {
std::vector<int32_t> local_output_padding(conv_transpose_attrs_.output_padding.begin(), conv_transpose_attrs_.output_padding.end());
if (conv_transpose_attrs_.kernel_shape_specified) {
for (size_t i = 0; i < kernel_shape.size() && i < kernel_shape_vec_size; ++i) {
local_kernel_shape.push_back(kernel_shape[i]);
local_kernel_shape.push_back(gsl::narrow_cast<int32_t>(kernel_shape[i]));
}
} else {
for (size_t i = 0; i < kernel_shape_vec_size; ++i) {
@ -71,13 +71,13 @@ class ConvTranspose : public JsKernel {
}
}
for (size_t i = 0; i < conv_transpose_attrs_.pads.size() && i < pads_vec_size; ++i) {
local_pads[i] = conv_transpose_attrs_.pads[i];
local_pads[i] = gsl::narrow_cast<int32_t>(conv_transpose_attrs_.pads[i]);
}
for (size_t i = 0; i < conv_transpose_attrs_.dilations.size() && i < dialations_vec_size; ++i) {
local_dilations[i] = conv_transpose_attrs_.dilations[i];
local_dilations[i] = gsl::narrow_cast<int32_t>(conv_transpose_attrs_.dilations[i]);
}
for (size_t i = 0; i < conv_transpose_attrs_.strides.size() && i < strides_vec_size; ++i) {
local_strides[i] = conv_transpose_attrs_.strides[i];
local_strides[i] = gsl::narrow_cast<int32_t>(conv_transpose_attrs_.strides[i]);
}
LOGS_DEFAULT(VERBOSE) << "output_shape = " << conv_transpose_attrs_.output_shape << std::endl;
LOGS_DEFAULT(VERBOSE) << "output_padding = " << conv_transpose_attrs_.output_padding << std::endl;

4
onnxruntime/core/providers/js/operators/split.h
View file

@ -27,7 +27,7 @@ class Split : public JsKernel, public SplitBase {
}
} else if (split_sizes_.size() == 0) {
// Compute split_sizes from input shape and num_outputs
auto total_split_size = info.node().InputDefs()[0]->Shape()->dim(axis_).dim_value();
auto total_split_size = info.node().InputDefs()[0]->Shape()->dim(gsl::narrow_cast<int32_t>(axis_)).dim_value();
int64_t split_size_sum = 0;
if (num_outputs_ < 0) {
num_outputs_ = info.node().OutputDefs().size();
@ -37,7 +37,7 @@ class Split : public JsKernel, public SplitBase {
num_outputs_, ")");
}
for (auto output : info.node().OutputDefs()) {
auto split_size = output->Shape()->dim(axis_).dim_value();
auto split_size = output->Shape()->dim(gsl::narrow_cast<int32_t>(axis_)).dim_value();
split_sizes.push_back(gsl::narrow_cast<int32_t>(split_size));
split_size_sum += split_size;
}