-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
Conversation
kazum
commented
Jun 26, 2020
- Update relay.ext.coremlcompiler based on the change in [RUNTIME] Introduce MetadataModule to separate code compilation/interpretation and weight initialization #5770.
- Support int32 for Core ML input and output
- Handle "run" in CoreMLRuntime::GetFunction to measure time with Module.time_evaluator.
- Return PackedFunc() when CoreMLRuntime::GetFunction handles nothing.
- Add test for each operator supported by CoreML codegen.
- Support relay.expand_dims and relay.nn.relu.
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 |
@zhiics Relevant changes to #5770 are:
We can at least pre-allocate memory. I think of doing more optimization for input tensor management in another PR :) |
@kazum no problem. We shouldn't do that in this PR. I was just curious if you have a plan to do that. |
Thanks @kazum |
* [CODEGEN][CONTRIB] Various update for CoreML codegen * fix lint error
* [CODEGEN][CONTRIB] Various update for CoreML codegen * fix lint error