[fix] fix typo in llm_cuda swiglu & swish op

OpenPPL · Jun 21, 2024 · bbbb45e · bbbb45e
1 parent dc9a5b4
commit bbbb45e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ppl/nn/engines/llm_cuda/ops/opmx/swiglu_op.cc b/src/ppl/nn/engines/llm_cuda/ops/opmx/swiglu_op.cc
@@ -55,7 +55,7 @@ KernelImpl* SwiGLUOp::CreateKernelImpl() const {
 ppl::common::RetCode SwiGLUOp::SerializeData(const ppl::nn::pmx::SerializationContext&, utils::DataStream* ds) const {
     flatbuffers::FlatBufferBuilder builder;
     auto fb_param = opmx::CreateSwishParam(builder, param_.get()->beta);
-    auto fb_op_param = opmx::CreateOpParam(builder, pmx::OpParamType_SwishParam, fb_param.Union());
+    auto fb_op_param = opmx::CreateOpParam(builder, opmx::OpParamType_SwishParam, fb_param.Union());
     opmx::FinishOpParamBuffer(builder, fb_op_param);
     return ds->Write(builder.GetBufferPointer(), builder.GetSize());
 }

diff --git a/src/ppl/nn/engines/llm_cuda/ops/opmx/swish_op.cc b/src/ppl/nn/engines/llm_cuda/ops/opmx/swish_op.cc
@@ -54,7 +54,7 @@ KernelImpl* SwishOp::CreateKernelImpl() const {
 ppl::common::RetCode SwishOp::SerializeData(const ppl::nn::pmx::SerializationContext&, utils::DataStream* ds) const {
     flatbuffers::FlatBufferBuilder builder;
     auto fb_param = opmx::CreateSwishParam(builder, param_.get()->beta);
-    auto fb_op_param = opmx::CreateOpParam(builder, pmx::OpParamType_SwishParam, fb_param.Union());
+    auto fb_op_param = opmx::CreateOpParam(builder, opmx::OpParamType_SwishParam, fb_param.Union());
     opmx::FinishOpParamBuffer(builder, fb_op_param);
     return ds->Write(builder.GetBufferPointer(), builder.GetSize());
 }