projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: d39a4ea) | patch
[Relay][Runtime] Add memory manager for NDArray (#3121)
authorJared Roesch <roeschinc@gmail.com>
Thu, 2 May 2019 16:10:34 +0000 (12:10 -0400)
committerHaichen Shen <shenhaichen@gmail.com>
Thu, 2 May 2019 16:10:34 +0000 (09:10 -0700)
commit83cb872e216cf4045f5ce123ba454a8a85c26e68
tree3eb5576f638cc093b385e14e1afc60c58d6c49a6tree | snapshot
parentd39a4ea000d6d2a1879c0a6a7aa819de4e16eb27commit | diff
[Relay][Runtime] Add memory manager for NDArray (#3121)

* Add support for custom NDArray memory management

Credit to @icemelon9 and @wweic

* Fix copy-paste issue

* Fix naive allocator.h

* Remove buffer field

* Apply Wei's suggestions.

Co-Authored-By: jroesch <roeschinc@gmail.com>
* Fix Wei's suggestion

* Fix go rts

* Break MM dependency

* Add docs and clean up diff

* Add more docs

* Move to VM folder

* Fix lint

* Remove Go dep.

* Rename to Empty

* Address Haichen's comments
golang/src/tvm_runtime_pack.cc diff | blob | history
include/tvm/runtime/ndarray.h diff | blob | history
src/runtime/vm/memory_manager.cc [new file with mode: 0644] blob
src/runtime/vm/memory_manager.h [new file with mode: 0644] blob
src/runtime/vm/naive_allocator.h [new file with mode: 0644] blob
src/runtime/vm/pooled_allocator.h [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;