mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-24 19:20:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #26043 from Abdurrahheem:ash/fix-nll-layer

Browse Source 
Support Matrices with axes > 6 CUDA
This commit is contained in:
Alexander Smorkalov 2024-08-21 11:12:43 +03:00 committed by GitHub
commit 9dcac41c66
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/dnn/src/cuda4dnn/csl/tensor.hpp
View File

@ -28,7 +28,7 @@
#include <utility>
#ifndef CSL_MAX_TENSOR_RANK
#define CSL_MAX_TENSOR_RANK 6
#define CSL_MAX_TENSOR_RANK 7
#endif
namespace cv { namespace dnn { namespace cuda4dnn { namespace csl {

1
modules/dnn/test/test_onnx_conformance_layer_filter__cuda_denylist.inl.hpp
View File

@ -90,7 +90,6 @@
"test_scatternd_max",
"test_scatternd_min",
"test_scatternd_multiply",
"test_nllloss_NCd1d2d3d4d5_none_no_weight_expanded", // crash: https://github.com/opencv/opencv/issues/25471
"test_dequantizelinear_blocked", // Issue https://github.com/opencv/opencv/issues/25999
"test_quantizelinear", // Issue https://github.com/opencv/opencv/issues/25999
"test_quantizelinear_axis", // Issue https://github.com/opencv/opencv/issues/25999