Tianqi Chen

Results 36 issues of Tianqi Chen

This is a cross ref RFC on DLPack based exchange. As of now, DLPack exchange relies on python functions such as `tensor.__dlpack__()`. While they works well for common cases, the...

RFC
API extension
topic: DLPack
topic: Device Handling
Needs Discussion

Previously we landed stream=None mapping to legacy default stream (a safer case). As DLPack get popularized, one most canonical use-case is to exchange between library and pytorch. As most libraries...

API change
topic: DLPack

## Background As of now we have standard interface for exchanging tensors through DLPack. One minor but still useful thing is to also provide a mechanism for solutions. Motivating example...

RFC
API extension
topic: DLPack
topic: Device Handling
Needs Discussion

This PR updates the exchange API to be capsule following the new convention. Note that the exchange API python recommendation is not yet updated on the array-api side and so...

Background in https://github.com/dmlc/dlpack/issues/175 we introduced the C exchange API. In the original discussion we have considered proposals of PyCapsule as field and in dunder. We discussed the choice array-api meeting...

Cross ref https://github.com/data-apis/array-api/issues/972. It would be useful to think about DLPack-based protocol for dtype and device exchange.