From cdfc949a5edcd95f2749d8dba3e62d301c6f8dc4 Mon Sep 17 00:00:00 2001 From: Jiacheng Huang Date: Fri, 7 Aug 2026 19:56:42 +0800 Subject: [PATCH] refactor(ops): use canonical InfiniOps softmax API --- src/infinicore/ops/softmax/softmax_infiniops.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/infinicore/ops/softmax/softmax_infiniops.cc b/src/infinicore/ops/softmax/softmax_infiniops.cc index c0087e1f8..3765eeb86 100644 --- a/src/infinicore/ops/softmax/softmax_infiniops.cc +++ b/src/infinicore/ops/softmax/softmax_infiniops.cc @@ -3,7 +3,7 @@ #ifdef ENABLE_INFINIOPS_API #include "../infiniops_impl.hpp" -#include "base/softmax_infinilm.h" +#include "base/softmax.h" #include @@ -22,7 +22,7 @@ void calculate(Tensor output, Tensor input, int axis) { TensorMeta output_meta(output); TensorMeta input_meta(input); - infini::ops::SoftmaxInfinilm::Call( + infini::ops::Softmax::Call( handle, config, input_meta.tensor(input),