PaddlePaddle
diff --git a/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.cc
Copy file name to clipboard
-36Lines changed: 0 additions & 36 deletions b/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.cc
Copy file name to clipboard
-36Lines changed: 0 additions & 36 deletions
diff --git a/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.cu.cc
Copy file name to clipboard
-27Lines changed: 0 additions & 27 deletions b/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.cu.cc
Copy file name to clipboard
-27Lines changed: 0 additions & 27 deletions
diff --git a/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.h
Copy file name to clipboard
-68Lines changed: 0 additions & 68 deletions b/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.h
Copy file name to clipboard
-68Lines changed: 0 additions & 68 deletions
diff --git a/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.kps
Copy file name to clipboard
-43Lines changed: 0 additions & 43 deletions b/‎paddle/fluid/operators/collective/c_sync_calc_stream_op.kps
Copy file name to clipboard
-43Lines changed: 0 additions & 43 deletions
diff --git a/‎paddle/fluid/operators/collective/c_sync_calc_stream_op_xpu.cc
Copy file name to clipboard
-28Lines changed: 0 additions & 28 deletions b/‎paddle/fluid/operators/collective/c_sync_calc_stream_op_xpu.cc
Copy file name to clipboard
-28Lines changed: 0 additions & 28 deletions
diff --git a/‎paddle/fluid/pir/dialect/operator/utils/utils.cc
Copy file name to clipboard
-1Lines changed: 0 additions & 1 deletion b/‎paddle/fluid/pir/dialect/operator/utils/utils.cc
Copy file name to clipboard
-1Lines changed: 0 additions & 1 deletion
diff --git a/‎paddle/phi/kernels/gpu/sync_calc_stream_kernel.cu
Copy file name to clipboard
+26Lines changed: 26 additions & 0 deletions b/‎paddle/phi/kernels/gpu/sync_calc_stream_kernel.cu
Copy file name to clipboard
+26Lines changed: 26 additions & 0 deletions
diff --git a/‎paddle/phi/kernels/impl/sync_calc_stream_kernel_impl.h
Copy file name to clipboard
+42Lines changed: 42 additions & 0 deletions b/‎paddle/phi/kernels/impl/sync_calc_stream_kernel_impl.h
Copy file name to clipboard
+42Lines changed: 42 additions & 0 deletions
diff --git a/‎paddle/phi/kernels/kps/sync_calc_stream_kernel.kps
Copy file name to clipboard
+38Lines changed: 38 additions & 0 deletions b/‎paddle/phi/kernels/kps/sync_calc_stream_kernel.kps
Copy file name to clipboard
+38Lines changed: 38 additions & 0 deletions
diff --git a/‎paddle/phi/kernels/xpu/sync_calc_stream_kernel.cc
Copy file name to clipboard
+26Lines changed: 26 additions & 0 deletions b/‎paddle/phi/kernels/xpu/sync_calc_stream_kernel.cc
Copy file name to clipboard
+26Lines changed: 26 additions & 0 deletions
@@ -39,7 +39,6 @@ const std::unordered_set<std::string> LegacyOpList = {
     LoadCombineOp::name(),
     CConcatOp::name(),
     CBroadcast_Op::name(),
-    CSyncCalcStream_Op::name(),
     CSyncCommStream_Op::name(),
     DistributedPushSparseOp::name(),
     SendV2Op::name(),
 
@@ -0,0 +1,26 @@
+// Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/impl/sync_calc_stream_kernel_impl.h"
+
+PD_REGISTER_KERNEL(sync_calc_stream,
+                   GPU,
+                   ALL_LAYOUT,
+                   phi::SyncCalcStreamKernel,
+                   float,
+                   double,
+                   int,
+                   int64_t,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
@@ -0,0 +1,42 @@
+// Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include <string>
+
+#include "paddle/phi/core/dense_tensor.h"
+#include "paddle/phi/core/kernel_registry.h"
+
+namespace phi {
+
+template <typename T, typename Context>
+void SyncCalcStreamKernel(const Context &dev_ctx,
+                          const DenseTensor &x,
+                          DenseTensor *out) {
+#if (defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)) && !defined(_WIN32)
+  phi::backends::gpu::GpuStreamSync(dev_ctx.stream());
+#elif defined(PADDLE_WITH_XPU_BKCL)
+  auto place = dev_ctx.GetPlace();
+  PADDLE_ENFORCE_EQ(place.GetType() == phi::AllocationType::XPU,
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Sync stream op can run on xpu place only for now."));
+  dev_ctx.Wait();
+#else
+  PADDLE_THROW(
+      phi::errors::PreconditionNotMet("PaddlePaddle should compile with GPU."));
+#endif
+}
+}  // namespace phi
@@ -0,0 +1,38 @@
+// Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifdef PADDLE_WITH_XPU_KP
+
+// Please do not modify the following code
+#if defined(__CUDA_ARCH__)
+#undef __CUDA_ARCH__
+#endif
+
+#if defined(__CUDACC__)
+#undef __CUDACC__
+#endif
+
+#if defined(__CUDA__)
+#undef __CUDA__
+#endif
+
+#if defined(__NVCC__)
+#undef __NVCC__
+#endif
+
+#include "paddle/phi/kernels/impl/sync_calc_stream_kernel_impl.h"
+
+PD_REGISTER_KERNEL(
+    sync_calc_stream, KPS, ALL_LAYOUT, phi::SyncCalcStreamKernel, float) {}
+#endif
@@ -0,0 +1,26 @@
+// Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/kernels/impl/sync_calc_stream_kernel_impl.h"
+
+PD_REGISTER_KERNEL(sync_calc_stream,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::SyncCalcStreamKernel,
+                   float,
+                   double,
+                   int,
+                   int64_t,
+                   phi::dtype::bfloat16,
+                   phi::dtype::float16) {}