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 @@ -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 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/relu/relu_infiniops.cc b/src/infinicore/ops/relu/relu_infiniops.cc index 7890765fe..ba7c510ce 100644 --- a/src/infinicore/ops/relu/relu_infiniops.cc +++ b/src/infinicore/ops/relu/relu_infiniops.cc @@ -3,7 +3,7 @@ #ifdef ENABLE_INFINIOPS_API #include "../infiniops_impl.hpp" -#include "base/relu_infinilm.h" +#include "base/relu.h" namespace infinicore::op::relu_impl::infiniops { namespace { @@ -20,7 +20,7 @@ void calculate(Tensor output, Tensor input) { TensorMeta output_meta(output); TensorMeta input_meta(input); - infini::ops::ReluInfinilm::Call( + infini::ops::Relu::Call( handle, config, input_meta.tensor(input), 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/src/infinicore/ops/silu_and_mul/silu_and_mul_infiniops.cc b/src/infinicore/ops/silu_and_mul/silu_and_mul_infiniops.cc index aa1bf3413..4d4d2befc 100644 --- a/src/infinicore/ops/silu_and_mul/silu_and_mul_infiniops.cc +++ b/src/infinicore/ops/silu_and_mul/silu_and_mul_infiniops.cc @@ -3,7 +3,7 @@ #ifdef ENABLE_INFINIOPS_API #include "../infiniops_impl.hpp" -#include "base/silu_and_mul_infinilm.h" +#include "base/silu_and_mul.h" namespace infinicore::op::silu_and_mul_impl::infiniops { namespace { @@ -25,7 +25,7 @@ void run(void *planned_meta) { infini::ops::Handle handle; handle.set_stream(context::getStream()); infini::ops::Config config; - infini::ops::SiluAndMulInfinilm::Call( + infini::ops::SiluAndMul::Call( handle, config, planned->input.tensor(planned->input_tensor), planned->output.tensor(planned->output_tensor)); } diff --git a/submodules/InfiniOps b/submodules/InfiniOps index 1b9c37659..21b07ebcf 160000 --- a/submodules/InfiniOps +++ b/submodules/InfiniOps @@ -1 +1 @@ -Subproject commit 1b9c37659211eeaeedb48b9691f79e4c9e4412f9 +Subproject commit 21b07ebcfdb0f993d2f3b672a4e38788e489fb80