UpdateOperationParam

功能

更新Operation的Param参数。

定义

1

Status UpdateOperationParam(Operation *operation, const OpParam &opParam);

参数列表

返回值说明

规格约束

支持FillOperation、 SortOperation、TopkToppSamplingOperation（部分代码开放）、atb/train_op_params.h的参数和EventOperation。

使用示例

1. 创建Operation。

   1 2 3 4 5

   train::FastSoftMaxParam param; param.headNum = 3; param.qSeqLen = { 10, 20, 30 }; atb::Operation *op = nullptr; atb::Status st = atb::CreateOperation(param, &op);

2. 获取当前Operation的param。

   1 2	train::FastSoftMaxParam cloneParam; st = atb::CloneOperationParam(op, cloneParam);

3. 更新当前Operation的param。

   1 2	cloneParam.headNum = 66; st = atb::UpdateOperationParam(op, cloneParam);