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<int32_t>(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();
   }