From 5f039a6569783e5a6b15d227798bcae82f307834 Mon Sep 17 00:00:00 2001 From: Jiacheng Huang Date: Fri, 7 Aug 2026 20:02:57 +0800 Subject: [PATCH] refactor(ops): use InfiniOps Copy for rearrange --- src/infinicore/ops/rearrange/rearrange_infiniops.cc | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/infinicore/ops/rearrange/rearrange_infiniops.cc b/src/infinicore/ops/rearrange/rearrange_infiniops.cc index 24b275763..cd642c23e 100644 --- a/src/infinicore/ops/rearrange/rearrange_infiniops.cc +++ b/src/infinicore/ops/rearrange/rearrange_infiniops.cc @@ -3,7 +3,7 @@ #ifdef ENABLE_INFINIOPS_API #include "../infiniops_impl.hpp" -#include "base/rearrange_infinilm.h" +#include "base/copy.h" namespace infinicore::op::rearrange_impl::infiniops { namespace { @@ -25,8 +25,8 @@ void run(void *planned_meta) { infini::ops::Handle handle; handle.set_stream(context::getStream()); infini::ops::Config config; - infini::ops::RearrangeInfinilm::Call( - handle, config, planned->x.tensor(planned->x_tensor), planned->y.tensor(planned->y_tensor)); + infini::ops::Copy::Call( + handle, config, planned->x.tensor(planned->x_tensor), false, planned->y.tensor(planned->y_tensor)); } void cleanup(void **planned_meta_ptr) { @@ -34,8 +34,5 @@ void cleanup(void **planned_meta_ptr) { *planned_meta_ptr = nullptr; } -// ReArrange is used by Tensor::copy_from during test result conversion. Keep -// the InfiniCore/InfiniOp implementation active until this adapter gets the same -// CUDA shim treatment as Gemm. } // namespace infinicore::op::rearrange_impl::infiniops #endif