zihaomu commented on PR #585:
URL: https://github.com/apache/tvm-ffi/pull/585#issuecomment-4430532911
@tqchen The scope of this PR has shifted from fixing ROCm-specific Torch C
DLPack extension loading to making tvm-ffi avoid loading the optional
out-of-tree `torch-c-dlpack-ext` package when upstream PyTorch
already provides the C DLPack exchange API.
This is simpler because newer PyTorch ROCm builds already expose a working
`__dlpack_c_exchange_api__`. In that case, tvm-ffi should trust PyTorch and
return early instead of forcing a local extension
override. The local extension remains only as a compatibility fallback for
older PyTorch versions
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
