Skip to content

Commit

Permalink
style change for conv_transpose.cc
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffdaily committed Oct 6, 2023
1 parent 5a8e343 commit 00944f4
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
3 changes: 2 additions & 1 deletion onnxruntime/core/providers/cuda/nn/conv_transpose.cc
Original file line number Diff line number Diff line change
Expand Up @@ -102,8 +102,9 @@ Status ConvTranspose<T>::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<CudaT>()));
}

// Special case when there is a dim value of 0 in the shape.
// Return only after we have cached the following for subsequent runs :
Expand Down
3 changes: 2 additions & 1 deletion onnxruntime/core/providers/rocm/nn/conv_transpose.cc
Original file line number Diff line number Diff line change
Expand Up @@ -93,8 +93,9 @@ Status ConvTranspose<T>::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<HipT>()));
}

// Special case when there is a dim value of 0 in the shape.
// Return only after we have cached the following for subsequent runs :
Expand Down

0 comments on commit 00944f4

Please sign in to comment.