Andrew Reusch
929717a03a
* Fix stack overflow when partially-__init__ Node raises exception. * If a Node subclass raises an exception and ctypes is in use before __init_handle_by_constructor__ is called (or self.handle is otherwise set), a Python stack overflow could result. This is because the unset handle slot causes self.handle accesses to fallback on the getattr(self, 'handle') method, invoking NodeGetAttr. * Then I believe this causes an infinite loop. * The fix is to make Node.__getattr__ raise AttributeError for all attributes in __slots__, then make __del__ tolerant to missing self.handle. * I don't believe cython is affected because it implements a descriptor to access its underlying chandle and that shouldn't be unset. * black format * actually use handle instead of self.handle
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
# pylint: disable=invalid-name, unused-import
|
|
"""Runtime Object API"""
|
|
import ctypes
|
|
|
|
from tvm._ffi.base import _FFI_MODE, _RUNTIME_ONLY, check_call, _LIB, c_str
|
|
from tvm._ffi.runtime_ctypes import ObjectRValueRef
|
|
from . import _ffi_api, _ffi_node_api
|
|
|
|
try:
|
|
# pylint: disable=wrong-import-position,unused-import
|
|
if _FFI_MODE == "ctypes":
|
|
raise ImportError()
|
|
from tvm._ffi._cy3.core import _set_class_object, _set_class_object_generic
|
|
from tvm._ffi._cy3.core import ObjectBase, PyNativeObject
|
|
except (RuntimeError, ImportError) as error:
|
|
# pylint: disable=wrong-import-position,unused-import
|
|
if _FFI_MODE == "cython":
|
|
raise error
|
|
from tvm._ffi._ctypes.packed_func import _set_class_object, _set_class_object_generic
|
|
from tvm._ffi._ctypes.object import ObjectBase, PyNativeObject
|
|
|
|
|
|
def _new_object(cls):
|
|
"""Helper function for pickle"""
|
|
return cls.__new__(cls)
|
|
|
|
|
|
class Object(ObjectBase):
|
|
"""Base class for all tvm's runtime objects."""
|
|
|
|
__slots__ = []
|
|
|
|
def __repr__(self):
|
|
return _ffi_node_api.AsRepr(self)
|
|
|
|
def __dir__(self):
|
|
class_names = dir(self.__class__)
|
|
fnames = _ffi_node_api.NodeListAttrNames(self)
|
|
size = fnames(-1)
|
|
return sorted([fnames(i) for i in range(size)] + class_names)
|
|
|
|
def __getattr__(self, name):
|
|
if name in self.__slots__:
|
|
raise AttributeError(f"{name} is not set")
|
|
|
|
try:
|
|
return _ffi_node_api.NodeGetAttr(self, name)
|
|
except AttributeError:
|
|
raise AttributeError("%s has no attribute %s" % (str(type(self)), name))
|
|
|
|
def __hash__(self):
|
|
return _ffi_api.ObjectPtrHash(self)
|
|
|
|
def __eq__(self, other):
|
|
return self.same_as(other)
|
|
|
|
def __ne__(self, other):
|
|
return not self.__eq__(other)
|
|
|
|
def __reduce__(self):
|
|
cls = type(self)
|
|
return (_new_object, (cls,), self.__getstate__())
|
|
|
|
def __getstate__(self):
|
|
handle = self.handle
|
|
if handle is not None:
|
|
return {"handle": _ffi_node_api.SaveJSON(self)}
|
|
return {"handle": None}
|
|
|
|
def __setstate__(self, state):
|
|
# pylint: disable=assigning-non-slot, assignment-from-no-return
|
|
handle = state["handle"]
|
|
self.handle = None
|
|
if handle is not None:
|
|
self.__init_handle_by_constructor__(_ffi_node_api.LoadJSON, handle)
|
|
|
|
def _move(self):
|
|
"""Create an RValue reference to the object and mark the object as moved.
|
|
|
|
This is a advanced developer API that can be useful when passing an
|
|
unique reference to an Object that you no longer needed to a function.
|
|
|
|
A unique reference can trigger copy on write optimization that avoids
|
|
copy when we transform an object.
|
|
|
|
Note
|
|
----
|
|
All the reference of the object becomes invalid after it is moved.
|
|
Be very careful when using this feature.
|
|
|
|
Examples
|
|
--------
|
|
|
|
.. code-block:: python
|
|
|
|
x = tvm.tir.Var("x", "int32")
|
|
x0 = x
|
|
some_packed_func(x._move())
|
|
# both x0 and x will points to None after the function call.
|
|
|
|
Returns
|
|
-------
|
|
rvalue : The rvalue reference.
|
|
"""
|
|
return ObjectRValueRef(self)
|
|
|
|
|
|
_set_class_object(Object)
|