add share_external_data interface demo #219
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
jiweibo
reviewed
Feb 23, 2022
|
|
||
| for (size_t i = 0; i < out_data.size(); i += 100) { | ||
| LOG(INFO) << i << " : " << out_data[i] << std::endl; | ||
| } |
jiweibo
reviewed
Feb 23, 2022
| // 参数：shape - 数据 shape | ||
| // 参数：place - 数据的存放位置 | ||
| // 参数：layout - 数据格式，默认为 NCHW，当前仅支持 NCHW | ||
| // 返回：None |
There was a problem hiding this comment.
注意事项：
1、输入的情况，用户保证输入指针数据预测过程中有效
2、输出的情况，用户保证输出指针的数据长度大于等于模型的输出数据size
jiweibo
reviewed
Feb 23, 2022
| } | ||
|
|
||
| for (size_t i = 0; i < FLAGS_warmup; ++i) | ||
| CHECK(predictor->Run()); |
There was a problem hiding this comment.
handle ShareExternalData之后才能run
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
添加 ShareExternalData 接口的 demo
使用说明：
用户可以通过 ShareExternalData 接口自己管理输入和输出的数据指针。在推理过程中，需要保证数据指针不发生改动；对于输出的数据，用户需要知道数据的shape和位置(CPU/GPU上)