rocblas_axpy_ex Interface Reference#
hipfort_rocblas::rocblas_axpy_ex Interface Reference
BLAS EX API. More...
Public Member Functions | |
| integer(kind(rocblas_status_success)) function | rocblas_axpy_ex_ (handle, n, alpha, alpha_type, x, x_type, incx, y, y_type, incy, execution_type) |
Detailed Description
BLAS EX API.
axpy_ex computes constant alpha multiplied by vector x, plus vector y.
y := alpha * x + y
Currently supported datatypes are as follows:
| alpha_type | x_type | y_type | execution_type |
|---|---|---|---|
| bf16_r | bf16_r | bf16_r | f32_r |
| f32_r | bf16_r | bf16_r | f32_r |
| f16_r | f16_r | f16_r | f16_r |
| f16_r | f16_r | f16_r | f32_r |
| f32_r | f16_r | f16_r | f32_r |
| f32_r | f32_r | f32_r | f32_r |
| f64_r | f64_r | f64_r | f64_r |
| f32_c | f32_c | f32_c | f32_c |
| f64_c | f64_c | f64_c | f64_c |
- Parameters
-
[in] handle - [rocblas_handle] handle to the rocblas library context queue. [in] n - [rocblas_int] the number of elements in x and y. [in] alpha - device pointer or host pointer to specify the scalar alpha. [in] alpha_type - [rocblas_datatype] specifies the datatype of alpha. [in] x - device pointer storing vector x. [in] x_type - [rocblas_datatype] specifies the datatype of vector x. [in] incx - [rocblas_int] specifies the increment for the elements of x. [in,out] y - device pointer storing vector y. [in] y_type - [rocblas_datatype] specifies the datatype of vector y. [in] incy - [rocblas_int] specifies the increment for the elements of y. [in] execution_type - [rocblas_datatype] specifies the datatype of computation.
Member Function/Subroutine Documentation
◆ rocblas_axpy_ex_()
| integer(kind(rocblas_status_success)) function hipfort_rocblas::rocblas_axpy_ex::rocblas_axpy_ex_ | ( | type(c_ptr), value | handle, |
| integer(c_int), value | n, | ||
| type(c_ptr), value | alpha, | ||
| integer(kind(rocblas_datatype_f16_r)), value | alpha_type, | ||
| type(c_ptr), value | x, | ||
| integer(kind(rocblas_datatype_f16_r)), value | x_type, | ||
| integer(c_int), value | incx, | ||
| type(c_ptr), value | y, | ||
| integer(kind(rocblas_datatype_f16_r)), value | y_type, | ||
| integer(c_int), value | incy, | ||
| integer(kind(rocblas_datatype_f16_r)), value | execution_type | ||
| ) |
The documentation for this interface was generated from the following file: