lint

microsoft · Oct 27, 2023 · 128f87c · 128f87c
1 parent 0eb9330
commit 128f87c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 14 deletions.
diff --git a/onnxruntime/contrib_ops/cuda/collective/distributed_expand.cc b/onnxruntime/contrib_ops/cuda/collective/distributed_expand.cc
@@ -51,22 +51,23 @@ Status DistributedExpand<T>::ComputeInternal(OpKernelContext* context) const {
   TensorShapeVector original_output_dims{p_shape, p_shape + shape_tensor->Shape().Size()};
   TensorShape original_output_shape(original_output_dims);
   ORT_ENFORCE(
-    onnxruntime::cuda::ComputeOutputShape(
-      Node().Name(),
-      original_input_shape,
-      original_output_dims, original_output_shape).IsOK());
+      onnxruntime::cuda::ComputeOutputShape(
+          Node().Name(),
+          original_input_shape,
+          original_output_dims, original_output_shape)
+          .IsOK());
 
   // Compute local output shape.
   const auto local_output_shape = ComputeShardShape(original_output_shape, output_sharding_spec);
 
   auto output_tensor = context->Output(0, local_output_shape);
 
   return FuncExpand(
-    this,
-    context,
-    input_tensor,
-    shape_tensor,
-    output_tensor);
+      this,
+      context,
+      input_tensor,
+      shape_tensor,
+      output_tensor);
 }
 
 ONNX_OPERATOR_TYPED_KERNEL_EX(

diff --git a/onnxruntime/core/providers/cuda/tensor/expand.cc b/onnxruntime/core/providers/cuda/tensor/expand.cc
@@ -148,7 +148,6 @@ Status FuncExpand(
     const Tensor* input_data_tensor,
     const Tensor* /*input_shape_tensor*/,
     Tensor* output_tensor) {
-
   TensorShape output_shape = output_tensor->Shape();
 
 #ifdef ENABLE_STRIDED_TENSORS
@@ -203,10 +202,11 @@ std::unique_ptr<Tensor> FuncExpand(
   TensorShape output_shape(output_dims);
 
   ORT_ENFORCE(
-    ComputeOutputShape(
-      cuda_kernel->Node().Name(),
-      input_data_tensor->Shape(),
-      output_dims, output_shape).IsOK());
+      ComputeOutputShape(
+          cuda_kernel->Node().Name(),
+          input_data_tensor->Shape(),
+          output_dims, output_shape)
+          .IsOK());
 
   // Pre-allocate output.
   AllocatorPtr alloc;