ADD Cuda MultiplyWithStream (#1142)

cuda: ADD MultiplyWithStream
hybridgroup · Mar 30, 2024 · 318aa7c · 318aa7c
1 parent 0f64f63
commit 318aa7c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cuda/arithm.go b/cuda/arithm.go
@@ -221,11 +221,19 @@ func MinWithStream(src1, src2 GpuMat, dst *GpuMat, s Stream) {
 // Multiply computes a matrix-matrix or matrix-scalar multiplication.
 //
 // For further details, please see:
-// https://docs.opencv.org/master/d8/d34/group__cudaarithm__elem.html#ga124315aa226260841e25cc0b9ea99dc3
+// https://docs.opencv.org/master/d8/d34/group__cudaarithm__elem.html#ga497cc0615bf717e1e615143b56f00591
 func Multiply(src1, src2 GpuMat, dst *GpuMat) {
 	C.GpuMultiply(src1.p, src2.p, dst.p, nil)
 }
 
+// MultiplyWithStream computes a matrix-matrix or matrix-scalar multiplication.
+//
+// For further details, please see:
+// https://docs.opencv.org/master/d8/d34/group__cudaarithm__elem.html#ga497cc0615bf717e1e615143b56f00591
+func MultiplyWithStream(src1, src2 GpuMat, dst *GpuMat, s Stream) {
+	C.GpuMultiply(src1.p, src2.p, dst.p, s.p)
+}
+
 // Sqr computes a square value of each matrix element.
 //
 // For further details, please see: