Input arg modifier return maybe (#5453)
* modified SetInputArgModifyFn * Delete the CHECK changes in the assign_op.cpp file * Format * Modified the OutputArgModifyFn interface * add return * maybe error stack from CheckAndConstructOp to OutputArgModifier callback function * maybe error stack from CheckAndConstructOp to OutputArgModifier callback function * OutputArgModifier return maybe part_1 * maybe error stack from CheckAndConstructOp to OutputArgModifier callback function * input_arg_modifier return maybe * change lambda for JUST macro * fix conflicts Co-authored-by:aishangjj <702572275@qq.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/framework/user_op_registry.h 1 addition, 1 deletiononeflow/core/framework/user_op_registry.h
- oneflow/core/operator/user_op.cpp 2 additions, 2 deletionsoneflow/core/operator/user_op.cpp
- oneflow/user/kernels/stateful_local_opkernel.cpp 1 addition, 1 deletiononeflow/user/kernels/stateful_local_opkernel.cpp
- oneflow/user/ops/assign_op.cpp 6 additions, 5 deletionsoneflow/user/ops/assign_op.cpp
- oneflow/user/ops/batch_gather_op.cpp 3 additions, 2 deletionsoneflow/user/ops/batch_gather_op.cpp
- oneflow/user/ops/broadcast_like_op.cpp 3 additions, 2 deletionsoneflow/user/ops/broadcast_like_op.cpp
- oneflow/user/ops/cast_like_op.cpp 3 additions, 2 deletionsoneflow/user/ops/cast_like_op.cpp
- oneflow/user/ops/categorical_ordinal_encode_op.cpp 2 additions, 1 deletiononeflow/user/ops/categorical_ordinal_encode_op.cpp
- oneflow/user/ops/combined_margin_loss_op.cpp 2 additions, 1 deletiononeflow/user/ops/combined_margin_loss_op.cpp
- oneflow/user/ops/dim_gather_op.cpp 6 additions, 4 deletionsoneflow/user/ops/dim_gather_op.cpp
- oneflow/user/ops/dropout_op.cpp 2 additions, 1 deletiononeflow/user/ops/dropout_op.cpp
- oneflow/user/ops/dynamic_loss_scale_schedule_op.cpp 5 additions, 4 deletionsoneflow/user/ops/dynamic_loss_scale_schedule_op.cpp
- oneflow/user/ops/fake_quantization_op.cpp 4 additions, 3 deletionsoneflow/user/ops/fake_quantization_op.cpp
- oneflow/user/ops/fused_bias_add_op.cpp 3 additions, 2 deletionsoneflow/user/ops/fused_bias_add_op.cpp
- oneflow/user/ops/fused_scale_tril_softmax_mask_scale_op.cpp 3 additions, 2 deletionsoneflow/user/ops/fused_scale_tril_softmax_mask_scale_op.cpp
- oneflow/user/ops/gather_op.cpp 3 additions, 2 deletionsoneflow/user/ops/gather_op.cpp
- oneflow/user/ops/gpt_data_loader_op.cpp 4 additions, 3 deletionsoneflow/user/ops/gpt_data_loader_op.cpp
- oneflow/user/ops/image_preprocess_ops.cpp 3 additions, 2 deletionsoneflow/user/ops/image_preprocess_ops.cpp
- oneflow/user/ops/min_max_observer_op.cpp 3 additions, 2 deletionsoneflow/user/ops/min_max_observer_op.cpp
- oneflow/user/ops/model_update_ops.cpp 69 additions, 46 deletionsoneflow/user/ops/model_update_ops.cpp
Please register or sign in to comment