From a35c077bfed0fdaee491b95da6e94871ffe8bcf2 Mon Sep 17 00:00:00 2001 From: Jiacheng Huang Date: Fri, 7 Aug 2026 00:35:03 +0800 Subject: [PATCH] refactor(ops): migrate conv2d to canonical InfiniOps API --- src/infinicore/ops/conv2d/conv2d_infiniops.cc | 6 +++--- submodules/InfiniOps | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/infinicore/ops/conv2d/conv2d_infiniops.cc b/src/infinicore/ops/conv2d/conv2d_infiniops.cc index 3cdf0c63e..1c989ddb0 100644 --- a/src/infinicore/ops/conv2d/conv2d_infiniops.cc +++ b/src/infinicore/ops/conv2d/conv2d_infiniops.cc @@ -3,7 +3,7 @@ #ifdef ENABLE_INFINIOPS_API #include "../infiniops_impl.hpp" -#include "base/conv_infinilm.h" +#include "base/conv2d.h" #include #include @@ -48,14 +48,14 @@ void calculate(Tensor output, bias_meta.emplace(bias); } - infini::ops::ConvInfinilm::Call( + infini::ops::Conv2d::Call( handle, config, input_meta.tensor(input), weight_meta.tensor(weight), bias_meta ? std::optional{bias_meta->tensor(bias)} : std::nullopt, - toInt64Vector(pads, n), toInt64Vector(strides, n), + toInt64Vector(pads, n), toInt64Vector(dilations, n), int64_t{1}, output_meta.tensor(output)); diff --git a/submodules/InfiniOps b/submodules/InfiniOps index e733e325b..1ecce7832 160000 --- a/submodules/InfiniOps +++ b/submodules/InfiniOps @@ -1 +1 @@ -Subproject commit e733e325b4af776261419e4dc7e1f9386c05fa71 +Subproject commit 1ecce7832e218735597445589f8375a30c8fb173