From dcc8fe656b131d56e208caca1fbb51dd26c9a754 Mon Sep 17 00:00:00 2001 From: satyajandhyala Date: Fri, 24 Feb 2023 23:24:26 -0800 Subject: [PATCH] Fix type mismatch in CUDA Trilu op. (#12863) Added type cast to int64_t to avoid overflow errors/alerts. --- onnxruntime/core/providers/cuda/tensor/trilu.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/cuda/tensor/trilu.cc b/onnxruntime/core/providers/cuda/tensor/trilu.cc index 8da1d4a242..77dc33f362 100644 --- a/onnxruntime/core/providers/cuda/tensor/trilu.cc +++ b/onnxruntime/core/providers/cuda/tensor/trilu.cc @@ -34,12 +34,12 @@ Status Trilu::ComputeInternal(OpKernelContext* ctx) const { const Tensor& input = *input_ptr; const auto& shape = input.Shape(); const auto& input_dims = shape.GetDims(); - int32_t rank = gsl::narrow_cast(input_dims.size()); + auto rank = input_dims.size(); if (rank < 2) { return Status(ONNXRUNTIME, INVALID_ARGUMENT, "Input tensor should have a rank of at least 2"); } Tensor* output = ctx->Output(0, shape); - int64_t matrix_size = input_dims[rank - 1] * input_dims[rank - 2]; + auto matrix_size = input_dims[rank - 1] * input_dims[rank - 2]; if (matrix_size == 0) { return Status::OK(); }