From 37b8fb8be8f075558083cddb8fc49b19f4812f82 Mon Sep 17 00:00:00 2001
From: Jiacheng Huang <huangjiacheng0709@outlook.com>
Date: Fri, 7 Aug 2026 00:43:12 +0800
Subject: [PATCH] refactor(ops): migrate GELU and sigmoid to canonical
 InfiniOps APIs

---
 src/infinicore/ops/gelu/gelu_infiniops.cc         | 4 ++--
 src/infinicore/ops/gelutanh/gelutanh_infiniops.cc | 7 +++++--
 src/infinicore/ops/sigmoid/sigmoid_infiniops.cc   | 4 ++--
 submodules/InfiniOps                              | 2 +-
 4 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/infinicore/ops/gelu/gelu_infiniops.cc b/src/infinicore/ops/gelu/gelu_infiniops.cc
index de0b35fde..ca8b1a5f0 100644
--- a/src/infinicore/ops/gelu/gelu_infiniops.cc
+++ b/src/infinicore/ops/gelu/gelu_infiniops.cc
@@ -3,7 +3,7 @@
 #ifdef ENABLE_INFINIOPS_API
 #include "../infiniops_impl.hpp"
 
-#include "base/gelu_infinilm.h"
+#include "base/gelu.h"
 
 #include <string>
 
@@ -22,7 +22,7 @@ void calculate(Tensor output, Tensor input) {
 
     TensorMeta output_meta(output);
     TensorMeta input_meta(input);
-    infini::ops::GeluInfinilm::Call(
+    infini::ops::Gelu::Call(
         handle,
         config,
         input_meta.tensor(input),
diff --git a/src/infinicore/ops/gelutanh/gelutanh_infiniops.cc b/src/infinicore/ops/gelutanh/gelutanh_infiniops.cc
index e9faf4b35..cce7fc62e 100644
--- a/src/infinicore/ops/gelutanh/gelutanh_infiniops.cc
+++ b/src/infinicore/ops/gelutanh/gelutanh_infiniops.cc
@@ -3,7 +3,9 @@
 #ifdef ENABLE_INFINIOPS_API
 #include "../infiniops_impl.hpp"
 
-#include "base/gelutanh_infinilm.h"
+#include "base/gelu.h"
+
+#include <string>
 
 namespace infinicore::op::gelutanh_impl::infiniops {
 namespace {
@@ -20,10 +22,11 @@ void calculate(Tensor output, Tensor input) {
 
     TensorMeta output_meta(output);
     TensorMeta input_meta(input);
-    infini::ops::GelutanhInfinilm::Call(
+    infini::ops::Gelu::Call(
         handle,
         config,
         input_meta.tensor(input),
+        std::string{"tanh"},
         output_meta.tensor(output));
 }
 
diff --git a/src/infinicore/ops/sigmoid/sigmoid_infiniops.cc b/src/infinicore/ops/sigmoid/sigmoid_infiniops.cc
index 0327e6c7c..194fde6b3 100644
--- a/src/infinicore/ops/sigmoid/sigmoid_infiniops.cc
+++ b/src/infinicore/ops/sigmoid/sigmoid_infiniops.cc
@@ -3,7 +3,7 @@
 #ifdef ENABLE_INFINIOPS_API
 #include "../infiniops_impl.hpp"
 
-#include "base/sigmoid_infinilm.h"
+#include "base/sigmoid.h"
 
 namespace infinicore::op::sigmoid_impl::infiniops {
 namespace {
@@ -35,7 +35,7 @@ void run(void *planned_meta) {
     handle.set_stream(context::getStream());
     infini::ops::Config config;
 
-    infini::ops::SigmoidInfinilm::Call(
+    infini::ops::Sigmoid::Call(
         handle,
         config,
         planned->input.tensor(planned->input_tensor),
diff --git a/submodules/InfiniOps b/submodules/InfiniOps
index e733e325b..498c9aaaa 160000
--- a/submodules/InfiniOps
+++ b/submodules/InfiniOps
@@ -1 +1 @@
-Subproject commit e733e325b4af776261419e4dc7e1f9386c05fa71
+Subproject commit 498c9aaaa1681b0bb11e4a3b83904e636be81d0e