From b540c69eaa174e471acc2257b21ba9213803e876 Mon Sep 17 00:00:00 2001
From: Jiacheng Huang <huangjiacheng0709@outlook.com>
Date: Fri, 7 Aug 2026 20:04:18 +0800
Subject: [PATCH] refactor(ops): use InfiniOps Convolution for conv2d

---
 src/infinicore/ops/conv2d/conv2d_infiniops.cc | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/infinicore/ops/conv2d/conv2d_infiniops.cc b/src/infinicore/ops/conv2d/conv2d_infiniops.cc
index 3cdf0c63e..f1a048569 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/convolution.h"
 
 #include <optional>
 #include <vector>
@@ -48,15 +48,17 @@ void calculate(Tensor output,
         bias_meta.emplace(bias);
     }
 
-    infini::ops::ConvInfinilm::Call(
+    infini::ops::Convolution::Call(
         handle,
         config,
         input_meta.tensor(input),
         weight_meta.tensor(weight),
         bias_meta ? std::optional<infini::ops::Tensor>{bias_meta->tensor(bias)} : std::nullopt,
-        toInt64Vector(pads, n),
         toInt64Vector(strides, n),
+        toInt64Vector(pads, n),
         toInt64Vector(dilations, n),
+        false,
+        std::vector<int64_t>(n, 0),
         int64_t{1},
         output_meta.tensor(output));
 }