Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[VM] Minor refactor for C++ memory alloc #7413

Merged
merged 9 commits into from
Feb 6, 2021

Conversation

masahi
Copy link
Member

@masahi masahi commented Feb 5, 2021

Some leftover items from #7369, to be cleaner and safer.

Instead of getting a pointer to packed func and call it indirectly, just declare C++ op function in a header and call it directly.

cc @zhiics @jroesch

src/relay/op/tensor/tensor.h Outdated Show resolved Hide resolved
@masahi masahi force-pushed the vm-alloc-refactor branch 2 times, most recently from f0c7bc8 to 3cce3ad Compare February 6, 2021 00:21
@zhiics zhiics merged commit 1f846f0 into apache:main Feb 6, 2021
alexwong pushed a commit to alexwong/tvm that referenced this pull request Feb 11, 2021
* started moving things to header

* directly call InvokeTVMOp

* done all memory op

* also refactor AllocTensor

* declare Prod

* remove cached func for Add, Multiply, Divide

* lint fix

* revert test change

* remove tensor.h and declare Prod in pattern_utils.h
Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
* started moving things to header

* directly call InvokeTVMOp

* done all memory op

* also refactor AllocTensor

* declare Prod

* remove cached func for Add, Multiply, Divide

* lint fix

* revert test change

* remove tensor.h and declare Prod in pattern_utils.h
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
* started moving things to header

* directly call InvokeTVMOp

* done all memory op

* also refactor AllocTensor

* declare Prod

* remove cached func for Add, Multiply, Divide

* lint fix

* revert test change

* remove tensor.h and declare Prod in pattern_utils.h
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants