device_math Module Reference

Data Types
interface	device_cadd

Functions/Subroutines
subroutine, public	device_copy (a_d, b_d, n, strm)
	Copy a vector \( a = b \).
subroutine, public	device_masked_copy_0 (a_d, b_d, mask_d, n, n_mask, strm)
	Copy a masked vector \( a(mask) = b(mask) \).
subroutine, public	device_masked_gather_copy_0 (a_d, b_d, mask_d, n, n_mask, strm)
	Gather a masked vector \( a(i) = b(mask(i)) \).
subroutine, public	device_masked_scatter_copy_0 (a_d, b_d, mask_d, n, n_mask, strm)
	Scatter a masked vector \( a((mask(i)) = b(i) \).
subroutine, public	device_masked_atomic_reduction_0 (a_d, b_d, mask_d, n, n_mask, strm)
subroutine, public	device_cfill_mask (a_d, c, n, mask_d, n_mask, strm)
	Fill a constant to a masked vector. \( a_i = c, for i in mask \).
subroutine, public	device_rzero (a_d, n, strm)
	Zero a real vector.
subroutine, public	device_rone (a_d, n, strm)
	Set all elements to one.
subroutine, public	device_cmult (a_d, c, n, strm)
	Multiplication by constant c \( a = c \cdot a \).
subroutine, public	device_cmult2 (a_d, b_d, c, n, strm)
	Multiplication by constant c \( a = c \cdot b \).
subroutine, public	device_cdiv (a_d, c, n, strm)
	Division of constant c by array \( a = c / a \).
subroutine, public	device_cdiv2 (a_d, b_d, c, n, strm)
	Division of constant c by array \( a = c / b \).
subroutine	device_radd (a_d, c, n, strm)
	Add a scalar to vector \( a = a + s \).
subroutine, public	device_cadd2 (a_d, b_d, c, n, strm)
	Add a scalar to vector \( a = b + s \).
subroutine, public	device_cfill (a_d, c, n, strm)
	Set all elements to a constant c \( a = c \).
subroutine, public	device_add2 (a_d, b_d, n, strm)
	Vector addition \( a = a + b \).
subroutine, public	device_add4 (a_d, b_d, c_d, d_d, n, strm)
subroutine, public	device_add2s1 (a_d, b_d, c1, n, strm)
subroutine, public	device_add2s2 (a_d, b_d, c1, n, strm)
	Vector addition with scalar multiplication \( a = c_1 a + b \) (multiplication on first argument)
subroutine, public	device_addsqr2s2 (a_d, b_d, c1, n, strm)
	Returns \( a = a + c1 * (b * b )\).
subroutine, public	device_add3 (a_d, b_d, c_d, n, strm)
	Vector addition \( a = b + c \).
subroutine, public	device_add3s2 (a_d, b_d, c_d, c1, c2, n, strm)
	Returns \( a = c1 * b + c2 * c \).
subroutine, public	device_add4s3 (a_d, b_d, c_d, d_d, c1, c2, c3, n, strm)
	Returns \( a = c1 * b + c2 * c + c3 * d\).
subroutine, public	device_add5s4 (a_d, b_d, c_d, d_d, e_d, c1, c2, c3, c4, n, strm)
	Returns \( a = a + c1 * b + c2 * c + c3 * d + c4 * e\).
subroutine, public	device_invcol1 (a_d, n, strm)
	Invert a vector \( a = 1 / a \).
subroutine, public	device_invcol2 (a_d, b_d, n, strm)
	Vector division \( a = a / b \).
subroutine, public	device_invcol3 (a_d, b_d, c_d, n, strm)
	Vector division \( a = b / c \).
subroutine, public	device_col2 (a_d, b_d, n, strm)
	Vector multiplication \( a = a \cdot b \).
subroutine, public	device_col3 (a_d, b_d, c_d, n, strm)
	Vector multiplication with 3 vectors \( a = b \cdot c \).
subroutine, public	device_subcol3 (a_d, b_d, c_d, n, strm)
	Returns \( a = a - b*c \).
subroutine, public	device_sub2 (a_d, b_d, n, strm)
	Vector substraction \( a = a - b \).
subroutine, public	device_sub3 (a_d, b_d, c_d, n, strm)
	Vector subtraction \( a = b - c \).
subroutine, public	device_addcol3 (a_d, b_d, c_d, n, strm)
	Returns \( a = a + b*c \).
subroutine, public	device_addcol4 (a_d, b_d, c_d, d_d, n, strm)
	Returns \( a = a + b*c*d \).
subroutine, public	device_addcol3s2 (a_d, b_d, c_d, s, n, strm)
	Returns \( a = a + s(b*c) \).
subroutine, public	device_vdot3 (dot_d, u1_d, u2_d, u3_d, v1_d, v2_d, v3_d, n, strm)
	Compute a dot product \( dot = u \cdot v \) (3-d version) assuming vector components \( u = (u_1, u_2, u_3) \) etc.
subroutine, public	device_vcross (u1_d, u2_d, u3_d, v1_d, v2_d, v3_d, w1_d, w2_d, w3_d, n, strm)
	Compute a cross product \( u = v \times w \) (3-d version) assuming vector components \( u = (u_1, u_2, u_3) \) etc.
real(kind=rp) function, public	device_vlsc3 (u_d, v_d, w_d, n, strm)
	Compute multiplication sum \( dot = u \cdot v \cdot w \).
real(kind=rp) function, public	device_glsc3 (a_d, b_d, c_d, n, strm)
	Weighted inner product \( a^T b c \).
subroutine, public	device_glsc3_many (h, w_d, v_d_d, mult_d, j, n, strm)
subroutine, public	device_add2s2_many (y_d, x_d_d, a_d, j, n, strm)
real(kind=rp) function, public	device_glsc2 (a_d, b_d, n, strm)
	Weighted inner product \( a^T b \).
real(kind=rp) function, public	device_glsubnorm (a_d, b_d, n, strm)
	Returns the norm of the difference of two vectors \( \sqrt{(a-b)^T (a-b)} \).
real(kind=rp) function, public	device_glsum (a_d, n, strm)
	Sum a vector of length n.
subroutine, public	device_absval (a_d, n, strm)
subroutine, public	device_pwmax2 (a_d, b_d, n, strm)
	Compute the point-wise maximum of two vectors \( a_i = \max(a_i, b_i) \).
subroutine, public	device_pwmax3 (a_d, b_d, c_d, n, strm)
	Compute the point-wise maximum of two vectors \( a_i = \max(b_i, c_i) \).
subroutine, public	device_cpwmax2 (a_d, c, n, strm)
	Compute the point-wise maximum of a vector and a scalar \( a_i = \max(a_i, c) \).
subroutine, public	device_cpwmax3 (a_d, b_d, c, n, strm)
	Compute the point-wise maximum of a vector and a scalar \( a_i = \max(b_i, c) \).
subroutine, public	device_pwmin2 (a_d, b_d, n, strm)
	Compute the point-wise minimum of two vectors \( a_i = \min(a_i, b_i) \).
subroutine, public	device_pwmin3 (a_d, b_d, c_d, n, strm)
	Compute the point-wise minimum of two vectors \( a_i = \min(b_i, c_i) \).
subroutine, public	device_cpwmin2 (a_d, c, n, strm)
	Compute the point-wise minimum of a vector and a scalar \( a_i = \min(a_i, c) \).
subroutine, public	device_cpwmin3 (a_d, b_d, c, n, strm)
	Compute the point-wise minimum of a vector and a scalar \( a_i = \min(b_i, c) \).
subroutine	device_iadd (a_d, c, n, strm)
	Add an integer scalar to vector \( a = a + s \).

Function/Subroutine Documentation

device_absval()

subroutine, public device_math::device_absval	(	type(c_ptr)	a_d,
		integer, intent(in)	n,
		type(c_ptr), optional	strm
	)

Definition at line 1273 of file device_math.F90.

device_add2()

subroutine, public device_math::device_add2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 466 of file device_math.F90.

Here is the caller graph for this function:

device_add2s1()

subroutine, public device_math::device_add2s1	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp)	c1,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 516 of file device_math.F90.

Here is the caller graph for this function:

device_add2s2()

subroutine, public device_math::device_add2s2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp)	c1,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 544 of file device_math.F90.

Here is the caller graph for this function:

device_add2s2_many()

subroutine, public device_math::device_add2s2_many	(	type(c_ptr), value	y_d,
		type(c_ptr), value	x_d_d,
		type(c_ptr), value	a_d,
		integer(c_int)	j,
		integer(c_int)	n,
		type(c_ptr), optional	strm
	)

Definition at line 1145 of file device_math.F90.

device_add3()

subroutine, public device_math::device_add3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 598 of file device_math.F90.

Here is the caller graph for this function:

device_add3s2()

subroutine, public device_math::device_add3s2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		real(kind=rp)	c1,
		real(kind=rp)	c2,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 624 of file device_math.F90.

Here is the caller graph for this function:

device_add4()

subroutine, public device_math::device_add4	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		type(c_ptr)	d_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 491 of file device_math.F90.

Here is the caller graph for this function:

device_add4s3()

subroutine, public device_math::device_add4s3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		type(c_ptr)	d_d,
		real(kind=rp)	c1,
		real(kind=rp)	c2,
		real(kind=rp)	c3,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 651 of file device_math.F90.

Here is the caller graph for this function:

device_add5s4()

subroutine, public device_math::device_add5s4	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		type(c_ptr)	d_d,
		type(c_ptr)	e_d,
		real(kind=rp)	c1,
		real(kind=rp)	c2,
		real(kind=rp)	c3,
		real(kind=rp)	c4,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 678 of file device_math.F90.

device_addcol3()

subroutine, public device_math::device_addcol3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 911 of file device_math.F90.

Here is the caller graph for this function:

device_addcol3s2()

subroutine, public device_math::device_addcol3s2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		real(kind=rp)	s,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 963 of file device_math.F90.

Here is the caller graph for this function:

device_addcol4()

subroutine, public device_math::device_addcol4	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		type(c_ptr)	d_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 937 of file device_math.F90.

Here is the caller graph for this function:

device_addsqr2s2()

subroutine, public device_math::device_addsqr2s2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp)	c1,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 571 of file device_math.F90.

Here is the caller graph for this function:

device_cadd2()

subroutine, public device_math::device_cadd2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 411 of file device_math.F90.

Here is the caller graph for this function:

device_cdiv()

subroutine, public device_math::device_cdiv	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 334 of file device_math.F90.

device_cdiv2()

subroutine, public device_math::device_cdiv2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 359 of file device_math.F90.

device_cfill()

subroutine, public device_math::device_cfill	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 439 of file device_math.F90.

Here is the caller graph for this function:

device_cfill_mask()

subroutine, public device_math::device_cfill_mask	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr)	mask_d,
		integer	n_mask,
		type(c_ptr), optional	strm
	)

Definition at line 202 of file device_math.F90.

Here is the caller graph for this function:

device_cmult()

subroutine, public device_math::device_cmult	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 280 of file device_math.F90.

Here is the caller graph for this function:

device_cmult2()

subroutine, public device_math::device_cmult2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 307 of file device_math.F90.

Here is the caller graph for this function:

device_col2()

subroutine, public device_math::device_col2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 782 of file device_math.F90.

Here is the caller graph for this function:

device_col3()

subroutine, public device_math::device_col3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 807 of file device_math.F90.

Here is the caller graph for this function:

device_copy()

subroutine, public device_math::device_copy	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 72 of file device_math.F90.

Here is the caller graph for this function:

device_cpwmax2()

subroutine, public device_math::device_cpwmax2	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1359 of file device_math.F90.

Here is the caller graph for this function:

device_cpwmax3()

subroutine, public device_math::device_cpwmax3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1388 of file device_math.F90.

Here is the caller graph for this function:

device_cpwmin2()

subroutine, public device_math::device_cpwmin2	(	type(c_ptr)	a_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1475 of file device_math.F90.

Here is the caller graph for this function:

device_cpwmin3()

subroutine, public device_math::device_cpwmin3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		real(kind=rp), intent(in)	c,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1504 of file device_math.F90.

Here is the caller graph for this function:

device_glsc2()

real(kind=rp) function, public device_math::device_glsc2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1171 of file device_math.F90.

Here is the caller graph for this function:

device_glsc3()

real(kind=rp) function, public device_math::device_glsc3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1081 of file device_math.F90.

Here is the caller graph for this function:

device_glsc3_many()

subroutine, public device_math::device_glsc3_many	(	real(c_rp), dimension(j)	h,
		type(c_ptr), value	w_d,
		type(c_ptr), value	v_d_d,
		type(c_ptr), value	mult_d,
		integer(c_int)	j,
		integer(c_int)	n,
		type(c_ptr), optional	strm
	)

Definition at line 1113 of file device_math.F90.

device_glsubnorm()

real(kind=rp) function, public device_math::device_glsubnorm	(	type(c_ptr), intent(in)	a_d,
		type(c_ptr), intent(in)	b_d,
		integer, intent(in)	n,
		type(c_ptr), optional	strm
	)

Definition at line 1205 of file device_math.F90.

Here is the caller graph for this function:

device_glsum()

real(kind=rp) function, public device_math::device_glsum	(	type(c_ptr)	a_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1241 of file device_math.F90.

Here is the caller graph for this function:

device_iadd()

subroutine device_math::device_iadd ( type(c_ptr), intent(inout)  a_d, integer, intent(in)  c, integer, intent(in)  n, type(c_ptr), optional  strm  )

private

Definition at line 1535 of file device_math.F90.

device_invcol1()

subroutine, public device_math::device_invcol1	(	type(c_ptr)	a_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 705 of file device_math.F90.

Here is the caller graph for this function:

device_invcol2()

subroutine, public device_math::device_invcol2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 731 of file device_math.F90.

Here is the caller graph for this function:

device_invcol3()

subroutine, public device_math::device_invcol3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 757 of file device_math.F90.

Here is the caller graph for this function:

device_masked_atomic_reduction_0()

subroutine, public device_math::device_masked_atomic_reduction_0	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	mask_d,
		integer	n,
		integer	n_mask,
		type(c_ptr), optional	strm
	)

Definition at line 175 of file device_math.F90.

device_masked_copy_0()

subroutine, public device_math::device_masked_copy_0	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	mask_d,
		integer	n,
		integer	n_mask,
		type(c_ptr), optional	strm
	)

Definition at line 98 of file device_math.F90.

Here is the caller graph for this function:

device_masked_gather_copy_0()

subroutine, public device_math::device_masked_gather_copy_0	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	mask_d,
		integer	n,
		integer	n_mask,
		type(c_ptr), optional	strm
	)

Definition at line 124 of file device_math.F90.

Here is the caller graph for this function:

device_masked_scatter_copy_0()

subroutine, public device_math::device_masked_scatter_copy_0	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	mask_d,
		integer	n,
		integer	n_mask,
		type(c_ptr), optional	strm
	)

Definition at line 150 of file device_math.F90.

Here is the caller graph for this function:

device_pwmax2()

subroutine, public device_math::device_pwmax2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1304 of file device_math.F90.

Here is the caller graph for this function:

device_pwmax3()

subroutine, public device_math::device_pwmax3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1331 of file device_math.F90.

Here is the caller graph for this function:

device_pwmin2()

subroutine, public device_math::device_pwmin2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1420 of file device_math.F90.

Here is the caller graph for this function:

device_pwmin3()

subroutine, public device_math::device_pwmin3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1447 of file device_math.F90.

Here is the caller graph for this function:

device_radd()

subroutine device_math::device_radd ( type(c_ptr)  a_d, real(kind=rp), intent(in)  c, integer  n, type(c_ptr), optional  strm  )

private

Definition at line 384 of file device_math.F90.

device_rone()

subroutine, public device_math::device_rone	(	type(c_ptr)	a_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 257 of file device_math.F90.

Here is the call graph for this function:

Here is the caller graph for this function:

device_rzero()

subroutine, public device_math::device_rzero	(	type(c_ptr)	a_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 231 of file device_math.F90.

Here is the caller graph for this function:

device_sub2()

subroutine, public device_math::device_sub2	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 859 of file device_math.F90.

Here is the caller graph for this function:

device_sub3()

subroutine, public device_math::device_sub3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 885 of file device_math.F90.

Here is the caller graph for this function:

device_subcol3()

subroutine, public device_math::device_subcol3	(	type(c_ptr)	a_d,
		type(c_ptr)	b_d,
		type(c_ptr)	c_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 833 of file device_math.F90.

Here is the caller graph for this function:

device_vcross()

subroutine, public device_math::device_vcross	(	type(c_ptr)	u1_d,
		type(c_ptr)	u2_d,
		type(c_ptr)	u3_d,
		type(c_ptr)	v1_d,
		type(c_ptr)	v2_d,
		type(c_ptr)	v3_d,
		type(c_ptr)	w1_d,
		type(c_ptr)	w2_d,
		type(c_ptr)	w3_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1018 of file device_math.F90.

Here is the caller graph for this function:

device_vdot3()

subroutine, public device_math::device_vdot3	(	type(c_ptr)	dot_d,
		type(c_ptr)	u1_d,
		type(c_ptr)	u2_d,
		type(c_ptr)	u3_d,
		type(c_ptr)	v1_d,
		type(c_ptr)	v2_d,
		type(c_ptr)	v3_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 991 of file device_math.F90.

Here is the caller graph for this function:

device_vlsc3()

real(kind=rp) function, public device_math::device_vlsc3	(	type(c_ptr)	u_d,
		type(c_ptr)	v_d,
		type(c_ptr)	w_d,
		integer	n,
		type(c_ptr), optional	strm
	)

Definition at line 1051 of file device_math.F90.

Here is the caller graph for this function: