Skip to content
forked from dmlc/dlpack

RFC for common tensor and operator guideline in deep learning system

License

Notifications You must be signed in to change notification settings

Yangqing/dlpack

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DLPack

RFC for common tensor and operator guidelines in deep learning systems. Started by this issue

Currently each DL framework have their own internal tensor data structure, and operator interface, which are similar but not compatible. The major goal of thie RFC is to come up with a minimum tensor data structure and a set of operator interfaces(or guidelines) that can be shared across frameworks.

This enables:

  • Easier sharing of operators between deep learning frameworks.
  • Easier wrapping of vendor level operator implementations, allowing collaboration when introducing new devices/ops.
  • Quick swapping of backend implementations, like different version of BLAS
  • For final users, this could bring more operators, and possiblity of mixing usage between frameworks (e.g. pass a Torch Tensor into Caffe, use MXNet scheduler to schedule Torch Tensors)

We do not intend to implement of Tensor and Ops, but instead use this as common interface to reuse tensor and ops across frameworks.

Credit

The initial interface so far base on discussions between @soumith @piiswrong @Yangqing @naibaf7 @bhack @edgarriba @tqchen

The future interface discussion will be based on issues on this repo.

About

RFC for common tensor and operator guideline in deep learning system

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C 37.3%
  • CMake 29.4%
  • C++ 25.4%
  • Makefile 7.9%