From 86bec7263f6fb00064b8fae89761264d46905d93 Mon Sep 17 00:00:00 2001 From: Jiacheng Huang Date: Fri, 7 Aug 2026 00:47:04 +0800 Subject: [PATCH] refactor(ops): migrate softmax to canonical InfiniOps API --- src/infinicore/ops/softmax/softmax_infiniops.cc | 4 ++-- submodules/InfiniOps | 2 +- 2 files changed, 3 insertions(+), 3 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), diff --git a/submodules/InfiniOps b/submodules/InfiniOps index e733e325b..eec65a135 160000 --- a/submodules/InfiniOps +++ b/submodules/InfiniOps @@ -1 +1 @@ -Subproject commit e733e325b4af776261419e4dc7e1f9386c05fa71 +Subproject commit eec65a135de32a430bf5a9efc132c3001aeafa9b