feat: add cuda_GpuMat to big types

This patch enables passing GpuMat as an in/out argument in several functions.
This commit is contained in:
Vadim Levin 2023-07-21 17:50:48 +03:00 committed by Alexander Smorkalov
parent e5e1a3bfde
commit be29c99d5a

View File

@ -511,7 +511,10 @@ class ArgInfo(object):
return self.enclosing_arg.name + '.' + self.name
def isbig(self):
return self.tp in ["Mat", "vector_Mat", "cuda::GpuMat", "GpuMat", "vector_GpuMat", "UMat", "vector_UMat"] # or self.tp.startswith("vector")
return self.tp in ["Mat", "vector_Mat",
"cuda::GpuMat", "cuda_GpuMat", "GpuMat",
"vector_GpuMat", "vector_cuda_GpuMat",
"UMat", "vector_UMat"] # or self.tp.startswith("vector")
def crepr(self):
return "ArgInfo(\"%s\", %d)" % (self.name, self.outputarg)