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

[CODEGEN][CONTRIB] Various update for CoreML codegen #5934

Merged
merged 2 commits into from Jun 26, 2020

Conversation

kazum
Copy link
Contributor

@kazum kazum commented Jun 26, 2020

@zhiics
Copy link
Member

zhiics commented Jun 26, 2020

aah, I haven't noticed that CoreML codegen was also using BYOC infra. Thanks for updating it. I have a question which is not relevant to this change. I see SetInput needs to allocate memory and copy from DLTensor to the local memory every time. Is it possible to pre-allocate memory or directly do zero-copy by sharing the passed tensor?

@kazum
Copy link
Contributor Author

kazum commented Jun 26, 2020

@zhiics Relevant changes to #5770 are:
https://github.com/apache/incubator-tvm/pull/5934/files#diff-639424334c798703f0e62ec8a5eaf779R30
https://github.com/apache/incubator-tvm/pull/5934/files#diff-271e7167e72d0f1de3aee097fa5cb5d2R232-R245

Is it possible to pre-allocate memory or directly do zero-copy by sharing the passed tensor?

We can at least pre-allocate memory. I think of doing more optimization for input tensor management in another PR :)

@zhiics
Copy link
Member

zhiics commented Jun 26, 2020

@kazum no problem. We shouldn't do that in this PR. I was just curious if you have a plan to do that.

@zhiics zhiics merged commit 69313a7 into apache:master Jun 26, 2020
@zhiics
Copy link
Member

zhiics commented Jun 26, 2020

Thanks @kazum

@kazum kazum deleted the coreml_codegen_update branch June 26, 2020 19:06
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Jun 30, 2020
* [CODEGEN][CONTRIB] Various update for CoreML codegen

* fix lint error
zhiics pushed a commit to neo-ai/tvm that referenced this pull request Jul 2, 2020
* [CODEGEN][CONTRIB] Various update for CoreML codegen

* fix lint error
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants