parameters: (self, stage)
return_statement: return self
docstring: Add stage id to primitive attribute.
  
          Note:
              It is valid only in semi auto parallel.
              In other parallel modes, please set it to be 0.
          Args:
              stage (int): The stage id for the current operation.
          Examples:
              >>> from mindspore import ops
              >>> add = ops.Add()
              >>> print(add.set_stage(0))
              Prim[Add]<stage=0>
docstring_summary: Add stage id to primitive attribute.
docstring_tokens: Add stage id to primitive attribute .
function: def set_stage(self, stage):
          """
          Add stage id to primitive attribute.
  
          Note:
              It is valid only in semi auto parallel.
              In other parallel modes, please set it to be 0.
          Args:
              stage (int): The stage id for the current operation.
          Examples:
              >>> from mindspore import ops
              >>> add = ops.Add()
              >>> print(add.set_stage(0))
              Prim[Add]<stage=0>
          """
          self.add_prim_attr("stage", stage)
          return self
function_tokens: def set_stage ( self , stage ) : self . add_prim_attr ( "stage" , stage ) return self