def _InitOpNodeSignatureDescSymbol(self, symbol_id, op_node_signature):
instruction = instr_util.InstructionProto()
instruction.instr_type_name = "InitOpNodeSignatureDescSymbol"
instruction.operand.append(_InitSymbolOperand(symbol_id))
self.instruction_list_.instruction.append(instruction)
eager_symbol = eager_symbol_util.EagerSymbol()
eager_symbol.symbol_id = symbol_id
eager_symbol.op_node_signature_symbol.CopyFrom(op_node_signature)
self.eager_symbol_list_.eager_symbol.append(eager_symbol)
def _InitOpConfSymbol(self, symbol_id, op_conf):
instruction = instr_util.InstructionProto()
instruction.instr_type_name = "InitOperatorConfSymbol"
instruction.operand.append(_InitSymbolOperand(symbol_id))
self.instruction_list_.instruction.append(instruction)
eager_symbol = eager_symbol_util.EagerSymbol()
eager_symbol.symbol_id = symbol_id
eager_symbol.op_conf_symbol.CopyFrom(op_conf)
self.eager_symbol_list_.eager_symbol.append(eager_symbol)
def _NewScopeSymbol(self, scope_proto):
instruction = instr_util.InstructionProto()
instruction.instr_type_name = "InitScopeSymbol"
instruction.operand.append(_InitSymbolOperand(scope_proto.symbol_id))
self.instruction_list_.instruction.append(instruction)
eager_symbol = eager_symbol_util.EagerSymbol()
eager_symbol.symbol_id = scope_proto.symbol_id
eager_symbol.scope_symbol.CopyFrom(scope_proto)
self.eager_symbol_list_.eager_symbol.append(eager_symbol)
def _NewParallelConfSymbol(self, symbol_id, parallel_conf):
instruction = instr_util.InstructionProto()
instruction.instr_type_name = "NewParallelDescSymbol"
instruction.operand.append(_Int64Operand(symbol_id))
self.instruction_list_.instruction.append(instruction)
eager_symbol = eager_symbol_util.EagerSymbol()
eager_symbol.symbol_id = symbol_id
eager_symbol.parallel_conf_symbol.CopyFrom(parallel_conf)
self.eager_symbol_list_.eager_symbol.append(eager_symbol)
def _InitStringSymbol(self, symbol_id, string):
instruction = instr_util.InstructionProto()
instruction.instr_type_name = "InitStringSymbol"
instruction.operand.append(_InitSymbolOperand(symbol_id))
self.instruction_list_.instruction.append(instruction)
eager_symbol = eager_symbol_util.EagerSymbol()
eager_symbol.symbol_id = symbol_id
eager_symbol.string_symbol = string
self.eager_symbol_list_.eager_symbol.append(eager_symbol)
def _InitOpConfSymbol(self, symbol_id, op_conf):
instruction = instr_cfg.InstructionProto()
instruction.set_instr_type_name("InitOperatorConfSymbol")
instruction.mutable_operand().Add().CopyFrom(
oneflow_api.deprecated.vm.InitSymbolOperand(symbol_id)
)
self.instruction_list().mutable_instruction().Add().CopyFrom(instruction)
eager_symbol = eager_symbol_pb.EagerSymbol()
eager_symbol.symbol_id = symbol_id
eager_symbol.op_conf_symbol.CopyFrom(op_conf)
eager_symbol = oneflow_api.deprecated.MakeEagerSymbolByString(str(eager_symbol))
self.eager_symbol_list().mutable_eager_symbol().Add().CopyFrom(eager_symbol)
