potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 07:43:03 +04:00
Code Packages Releases Wiki Activity

Solve the bug of same shape broadcast with CUDA

Browse Source
This commit is contained in:
wanli
2023-04-28 17:25:49 +08:00
committed by Wanli
parent 85b04f0b4d
commit 46991bcd62
3 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/cuda/eltwise_ops.cu
+4 -1
View File
@@ -150,7 +150,7 @@ void eltwise_op(const Stream& stream, TensorSpan<T> output, TensorView<T> x, Ten
*/
for (int r = 0; r < output.rank(); r++)
{
while (x.get_axis_size(r) == 1 && y.get_axis_size(r) == 1) {
while (x.rank() > r && y.rank() > r && x.get_axis_size(r) == 1 && y.get_axis_size(r) == 1) {
CV_Assert(output.get_axis_size(r) == 1);
x.squeeze(r);
@@ -183,6 +183,9 @@ void eltwise_op(const Stream& stream, TensorSpan<T> output, TensorView<T> x, Ten
auto new_size = inShape1[i] * inShape1[j];
inShape1[i] = new_size;
inShape2[i] = new_size;
// outShape should be changed after merged
auto output_new_size = outShape[i] * outShape[j];
outShape[i] = output_new_size;
/* delete axis `j` */
inShape1.erase(std::begin(inShape1) + j);
modules/dnn/src/layers/nary_eltwise_layers.cpp
-2
View File
@@ -51,8 +51,6 @@ public:
WHERE,
} op;
// If the eltwise implementation is modified, you need to force enable the 'Layer_Test_Eltwise_bcast'
// test in the 'test_layers.cpp' file to make sure it all passes
NaryEltwiseLayerImpl(const LayerParams& params)
{
setParamsFrom(params);