[Autoscan] add 0D-tensor test for CPU #10214
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
|
Thanks for your contribution! |
CMakeLists.txt
Outdated
| @@ -74,6 +74,7 @@ lite_option(LITE_ON_TINY_PUBLISH "Publish tiny predictor lib." | |||
| lite_option(LITE_ON_MODEL_OPTIMIZE_TOOL "Build the model optimize tool" OFF) | |||
| lite_option(LITE_WITH_BENCHMARK_TEST "Build benchmark test cases" OFF) | |||
| lite_option(LITE_THREAD_POOL "Enable thread pool in lite" OFF) | |||
| lite_option(LITE_BLOCK_0DIM_PASS "Block support_0_dim_tensor_pass" OFF) | |||
There was a problem hiding this comment.
LITE_SKIP_SUPPORT_0_DIM_TENSOR_PASS
cmake/configure.cmake
Outdated
| @@ -263,3 +263,7 @@ if (EMSCRIPTEN) | |||
| add_compile_options("-pthread") | |||
| set(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKER_FLAGS} -pthread") | |||
| endif() | |||
|
|
|||
| if(LITE_BLOCK_0DIM_PASS) | |||
| add_definitions("-DLITE_BLOCK_0DIM_PASS") | |||
| @@ -60,6 +60,15 @@ bool naive_not<bool>(bool a) { | |||
| return !a; | |||
| } | |||
|
|
|||
| #define PROCESS_0D \ | |||
There was a problem hiding this comment.
这个有点 trick，能直接改 GenBatchElementWiseArg 吗？
There was a problem hiding this comment.
这位置我最开始也是想改 GenBatchElementWiseArg。。。但是改完发现有内存错误，所以想先绕过这块，等之后再修复可以吗
lite/operators/gaussian_random_op.cc
Outdated
| if (param_.ShapeTensor == nullptr && param_.ShapeTensorList.empty()) { | ||
| CHECK(param_.shape.size() > 0) | ||
| << "Attribute(shape) of GaussRandomOp must be set and shape.size() > 0"; | ||
| if (param_.ShapeTensor == nullptr && param_.ShapeTensorList.empty() && |
| @@ -102,6 +103,9 @@ | |||
| args = parser.parse_args() | |||
|
|
|||
| logging.basicConfig(level=logging.INFO, format="%(message)s") | |||
| logg = logging.getLogger(__name__) | |||
There was a problem hiding this comment.
用logger 或 log 会不会更好？
| from functools import partial | ||
|
|
||
|
|
||
| def check_broadcast(x_shape, y_shape, axis): |
There was a problem hiding this comment.
这个是不是可以抽出来作为公共模块？我理解大部分eltwise 类算子都需要使用的。
There was a problem hiding this comment.
这个确实可以复用，但应该放哪个文件里，就一个函数新建个文件是不是有点浪费
There was a problem hiding this comment.
mjp9527
commented
Apr 21, 2023
| if len(axes) == 0 or len(in_shape) <= 1: | ||
| return True | ||
|
|
||
| self.add_ignore_check_case( |
mjp9527
commented
Apr 21, 2023
| echo "Don't have paddlepaddle, need install." | ||
| fi | ||
| } | ||
|
|
There was a problem hiding this comment.
更新paddle可以成功修复x86/Host AutoScan
hong19860320
approved these changes
Apr 23, 2023
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.
PR devices
Arm/Host
PR types
New features
PR changes
OP
Description
（1）增加android/linux编译选项 block_0dim_pass=ON/OFF，用来判断是否开启support_0_dim_pass;
（2）为AutoScan单测增加0D测试case，同时增加 add_ignore_pass_case，目前只测Host/Arm CPU；
（3）修复部分算子不支持0D的问题；