From 585b93a10d6532b7a9c172bca5a8af61cc6f2471 Mon Sep 17 00:00:00 2001
From: Wang Yi <53533850+marigoold@users.noreply.github.com>
Date: Fri, 11 Aug 2023 20:35:38 +0800
Subject: [PATCH] Update flip_kernel.cu

---
 oneflow/user/kernels/flip_kernel.cu | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/oneflow/user/kernels/flip_kernel.cu b/oneflow/user/kernels/flip_kernel.cu
index b9d48957928..cafdfe0bc3b 100644
--- a/oneflow/user/kernels/flip_kernel.cu
+++ b/oneflow/user/kernels/flip_kernel.cu
@@ -90,12 +90,12 @@ class FlipGpuKernel final : public user_op::OpKernel {
     for (auto x : dims) { vis.val[x] = true; }
 
     if (dims.size() == 1 && dims[0] == x_tensor->shape_view().NumAxes() - 1) {
-      //       FlipLastDimGpuForward<T><<<BlocksNum4ThreadsNum(elem_cnt), kCudaThreadsNumPerBlock,
-      //       0,
-      //                                  ctx->stream()->As<ep::CudaStream>()->cuda_stream()>>>(
-      //           elem_cnt, x_tensor->shape_view().At(total_dims - 1), x_tensor->dptr<T>(),
-      //           y_tensor->mut_dptr<T>());
-      //       return;
+        FlipLastDimGpuForward<T><<<BlocksNum4ThreadsNum(elem_cnt), kCudaThreadsNumPerBlock,
+       0,
+                                  ctx->stream()->As<ep::CudaStream>()->cuda_stream()>>>(
+           elem_cnt, x_tensor->shape_view().At(total_dims - 1), x_tensor->dptr<T>(),
+           y_tensor->mut_dptr<T>());
+       return;
     }
 
     SIZE_V sizes_v;