DLPack Conversion API (#1573) (f52255b9) · Commits · cld / ml / tvm · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
f52255b9

Commit f52255b9 authored 6 years ago by eqy Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

DLPack Conversion API (#1573)

parent b0368e33

No related branches found

No related tags found

No related merge requests found

Changes 7

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/runtime/c_runtime_api.h 26 additions, 0 deletions

include/tvm/runtime/c_runtime_api.h
include/tvm/runtime/ndarray.h 1 addition, 1 deletion

include/tvm/runtime/ndarray.h
python/tvm/_ffi/ndarray.py 61 additions, 1 deletion

python/tvm/_ffi/ndarray.py
python/tvm/contrib/dlpack.py 43 additions, 0 deletions

python/tvm/contrib/dlpack.py
python/tvm/ndarray.py 1 addition, 1 deletion

python/tvm/ndarray.py
src/runtime/ndarray.cc 29 additions, 7 deletions

src/runtime/ndarray.cc
tests/python/contrib/test_dlpack.py 44 additions, 0 deletions

tests/python/contrib/test_dlpack.py

with 205 additions and 10 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment