DLPack:开放式内存张量结构
文档:https://dmlc.github.io/dlpack/latest
DLPack是一种开放式内存张量结构,用于在框架之间共享张量。DLPack可以实现:
- 更轻松地在深度学习框架之间共享操作符。
- 更容易封装供应商级别的操作符实现,在引入新设备/操作时促进协作。
- 快速切换后端实现,如不同版本的BLAS。
- 对于最终用户,这可以带来更多的操作符,以及在框架之间混合使用的可能性。
我们不打算实现张量和操作,而是将其作为跨框架复用张量和操作的通用桥梁。
提案程序
RFC提案以问题的形式提出。主要版本发布将通过投票问题进行,以确保参与者同意这些变更。
项目结构
目前有两个主要组成部分:
- include:稳定的头文件
- contrib:正在进行中的不稳定库
参与者
以下是参与DLPack RFC设计提案的人员列表:
@soumith @piiswrong @Yangqing @naibaf7 @bhack @edgarriba @tqchen @prigoyal @zdevito