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 index c66baabe4..1a7128315 100644 --- 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 index 8c945b9b5..294f6ec20 100644 --- 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()); }