colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add macro VSI_EXPAND_BROADCAST_ENABLE_DIMENSIONS for unit test compatibility (#386)

This commit is contained in:
Antkillerfarm 2022-05-06 09:30:26 +08:00 committed by GitHub
parent 11572140d2
commit c6847981e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tim/vx/ops/broadcast_test.cc
View File

@ -28,7 +28,9 @@
#include "gtest/gtest.h" #include "gtest/gtest.h"
#include "test_utils.h" #include "test_utils.h"
#include "vsi_nn_pub.h"
#ifdef VSI_EXPAND_BROADCAST_ENABLE_DIMENSIONS
static void CheckResult(std::shared_ptr<tim::vx::Graph>& graph, static void CheckResult(std::shared_ptr<tim::vx::Graph>& graph,
std::vector<float>& golden, std::vector<float>& golden,
std::shared_ptr<tim::vx::Tensor>& output_tensor) { std::shared_ptr<tim::vx::Tensor>& output_tensor) {
@ -327,3 +329,4 @@ TEST(Broadcast, 2DTo3D_WithDims12) {
CheckResult(graph, golden, output_tensor); CheckResult(graph, golden, output_tensor);
} }
#endif