saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix an arithmetic overflow warning (#12961)

Browse source
This commit is contained in:
Ye Wang 2022-09-15 15:53:57 -07:00 committed by GitHub
parent 739b5675c8
commit 3c427a8946
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/contrib_ops/cuda/bert/decoder_attention.cc
View file

@ -85,7 +85,7 @@ Status CheckInputs(const TensorShape& query_shape,
return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT, "q_weights shall have shape (hidden size, hidden size)");
}
if (kv_weights_dims[0] != hidden_size || kv_weights_dims[1] != static_cast<int64_t>(2 * hidden_size)) {
if (kv_weights_dims[0] != hidden_size || kv_weights_dims[1] != 2 * static_cast<int64_t>(hidden_size)) {
return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT, "kv_weights shall have shape (hidden size, 2 * hidden size)");
}
@ -95,7 +95,7 @@ Status CheckInputs(const TensorShape& query_shape,
bias_dims.size());
}
if (bias_dims[0] != 3 * hidden_size) {
if (bias_dims[0] != 3 * static_cast<int64_t>(hidden_size)) {
return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT, "bias shall have shape (3 * hidden size)");
}