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 #include @@ -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{bias_meta->tensor(bias)} : std::nullopt, - toInt64Vector(pads, n), toInt64Vector(strides, n), + toInt64Vector(pads, n), toInt64Vector(dilations, n), + false, + std::vector(n, 0), int64_t{1}, output_meta.tensor(output)); }