diff --git a/python/paddle/base/dygraph/tensor_patch_methods.py b/python/paddle/base/dygraph/tensor_patch_methods.py index 071324085893a1..200c69b8c31044 100644 --- a/python/paddle/base/dygraph/tensor_patch_methods.py +++ b/python/paddle/base/dygraph/tensor_patch_methods.py @@ -1601,6 +1601,7 @@ def _get_c_dlpack_exchange_api(): return core.dlpack_exchange_api_ptr() except Exception: pass + # For tvm ffi 0.1.4 only, in tvm ffi 0.1.5+, replaced by `__dlpack_c_exchange_api__` return core.dlpack_exchange_api_pycapsule() if not hasattr(core, "eager"): @@ -1652,7 +1653,9 @@ def _get_c_dlpack_exchange_api(): ("__dlpack_device__", __dlpack_device__), ("get_device", get_device), ("__tvm_ffi_env_stream__", __tvm_ffi_env_stream__), + # For TVM FFI 0.1.0-0.1.4, replaced by `__dlpack_c_exchange_api__` in TVM FFI 0.1.5+ ("__c_dlpack_exchange_api__", _get_c_dlpack_exchange_api()), + ("__dlpack_c_exchange_api__", core.dlpack_exchange_api_pycapsule()), ("device", device), ): setattr(core.eager.Tensor, method_name, method) diff --git a/python/unittest_py/requirements.txt b/python/unittest_py/requirements.txt index 67b7d301c69cb0..68851e93d319f3 100644 --- a/python/unittest_py/requirements.txt +++ b/python/unittest_py/requirements.txt @@ -19,6 +19,6 @@ xdoctest==1.3.0 ubelt==1.3.3 # just for xdoctest mypy==1.18.2 soundfile -apache-tvm-ffi==0.1.4 +apache-tvm-ffi==0.1.5 graphviz nvidia-ml-py3 ; platform_system != "Darwin" diff --git a/test/dygraph_to_static/test_tensor_attr_consistency.py b/test/dygraph_to_static/test_tensor_attr_consistency.py index b68c2db87fe609..93752d6aff7aa3 100644 --- a/test/dygraph_to_static/test_tensor_attr_consistency.py +++ b/test/dygraph_to_static/test_tensor_attr_consistency.py @@ -82,6 +82,7 @@ "__dlpack_device__", "__tvm_ffi_env_stream__", "__c_dlpack_exchange_api__", + "__dlpack_c_exchange_api__", ] ) STATIC_ONLY_TENSOR_ATTRS_ALLOW_LIST = OrderedSet(