"""ObjectRef"""
from yr.exception import YRInvokeError, YRError, GeneratorFinished
from yr.err_type import ErrorInfo, ErrorCode
import yr
from yr import log
from yr.common import constants
class NpuObject:
"""ObjectRef"""
def __init__(self, npu_id: str, size, dtype, device, ds_address, instance_id):
"""Initialize the ObjectRef."""
self._id = npu_id
self._size = size
self._dtype = dtype
self._device = device
self._ds_address = ds_address
self._instance_id = instance_id
def __del__(self):
pass
def __copy__(self):
return self
def __deepcopy__(self, memo):
return self
def __str__(self):
return self.id
def __eq__(self, other):
return self.id == other.id
def __hash__(self):
return hash(self.id)
def __repr__(self):
return f"NpuObject(id={self._id}, size={self._size}, dtype={self._dtype}, deviece={self._device})"
@property
def id(self):
"""npu_obj id."""
return self._id
@property
def size(self):
"""npu_obj size."""
return self._size
@property
def dtype(self):
"""npu_obj size."""
return self._dtype
@property
def instance_id(self):
"""npu_obj size."""
return self._instance_id