From e89ff869866225667ab250c834442914dde8a9c2 Mon Sep 17 00:00:00 2001 From: Jiacheng Huang Date: Fri, 7 Aug 2026 00:02:25 +0800 Subject: [PATCH] refactor(ops): migrate rearrange to canonical InfiniOps API --- src/infinicore/ops/rearrange/rearrange_infiniops.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/infinicore/ops/rearrange/rearrange_infiniops.cc b/src/infinicore/ops/rearrange/rearrange_infiniops.cc index 24b275763..28b8975ec 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) {