Merge pull request #24046 from VadimLevin:dev/vlevin/add-cuda-gpu-mat-to-big-types

feat: add cuda_GpuMat to big types
This commit is contained in:
Alexander Smorkalov 2023-07-26 13:18:32 +03:00 committed by GitHub
commit a25e809da1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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)