Fix mistake introcuded in previous PR and increase test coverage to avod this happening again

2019-12-16 18:38:58 +00:00
parent ba7b0f4c54
commit d427cebd12
2 changed files with 68 additions and 1 deletions
@@ -349,7 +349,7 @@ class ArgInfo(object):
        self.py_outputarg = False

    def isbig(self):
-        return self.tp in ["Mat", "vector_Mat", "GpuMat", "UMat", "vector_UMat"] # or self.tp.startswith("vector")
+        return self.tp in ["Mat", "vector_Mat", "cuda::GpuMat", "GpuMat", "vector_GpuMat", "UMat", "vector_UMat"] # or self.tp.startswith("vector")

    def crepr(self):
        return "ArgInfo(\"%s\", %d)" % (self.name, self.outputarg)