tqchen opened a new pull request, #131: URL: https://github.com/apache/tvm-ffi/pull/131
This PR refactors the TVMFFIEnvTensorAllocator to align with the finalized naming of DPack standard. - TVMFFIEnvSetTensorAllocator -> TVMFFIEnvSetDLPackManagedTensorAllocator - TVMFFIEnvGetTensorAllocator -> TVMFFIEnvGetDLPackManagedTensorAllocator We also introduced TVMFFIEnvTensorAlloc to directly allocate a ffi::Tensor from the given prototype which can be used by DSL compilers. Note that the new TVMFFIEnvTensorAlloc makes metadata allocation in libtvm_ffi so it won't suffer from the module unloading order problem. We removed Tensor::FromDLPackAlloc in favor of Tensor::FromEnvAlloc that makes use of TVMFFIEnvTensorAlloc. Hopefully this step allows us to stablize the tensor alloc api to align with dlpack before we freeze. -- 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]
