From 00944f43b0c665227721d503afa176edf89d0ab1 Mon Sep 17 00:00:00 2001 From: Jeff Daily Date: Fri, 6 Oct 2023 15:16:50 +0000 Subject: [PATCH] style change for conv_transpose.cc --- onnxruntime/core/providers/cuda/nn/conv_transpose.cc | 3 ++- onnxruntime/core/providers/rocm/nn/conv_transpose.cc | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/onnxruntime/core/providers/cuda/nn/conv_transpose.cc b/onnxruntime/core/providers/cuda/nn/conv_transpose.cc index 2b3326c528659..04f6bc46dcfcc 100644 --- a/onnxruntime/core/providers/cuda/nn/conv_transpose.cc +++ b/onnxruntime/core/providers/cuda/nn/conv_transpose.cc @@ -102,8 +102,9 @@ Status ConvTranspose::DoConvTranspose(OpKernelContext* context, bool dynamic_ } s_.y_dims = gsl::make_span(y_dims); - if (w_dims_changed) + if (w_dims_changed) { ORT_RETURN_IF_ERROR(s_.w_desc.Set(w_dims, CudnnTensor::GetDataType())); + } // Special case when there is a dim value of 0 in the shape. // Return only after we have cached the following for subsequent runs : diff --git a/onnxruntime/core/providers/rocm/nn/conv_transpose.cc b/onnxruntime/core/providers/rocm/nn/conv_transpose.cc index 23e9faedb1e76..475d26d2e306d 100644 --- a/onnxruntime/core/providers/rocm/nn/conv_transpose.cc +++ b/onnxruntime/core/providers/rocm/nn/conv_transpose.cc @@ -93,8 +93,9 @@ Status ConvTranspose::DoConvTranspose(OpKernelContext* context, bool dynamic_ } s_.y_dims = gsl::make_span(y_dims); - if (w_dims_changed) + if (w_dims_changed) { ORT_RETURN_IF_ERROR(s_.w_desc.Set(w_dims, MiopenTensor::GetDataType())); + } // Special case when there is a dim value of 0 in the shape. // Return only after we have cached the following for subsequent runs :