| 58 | |
| 59 | template<typename T> |
| 60 | void magmablas_laset(magma_uplo_t uplo, magma_int_t m, magma_int_t n, T offdiag, |
| 61 | T diag, cl_mem dA, size_t dA_offset, magma_int_t ldda, |
| 62 | magma_queue_t queue) { |
| 63 | using arrayfire::opencl::kernel::laset; |
| 64 | magma_int_t info = 0; |
| 65 | if (uplo != MagmaLower && uplo != MagmaUpper && uplo != MagmaFull) { |
| 66 | info = -1; |
| 67 | } else if (m < 0) { |
| 68 | info = -2; |
| 69 | } else if (n < 0) { |
| 70 | info = -3; |
| 71 | } else if (ldda < std::max(1, m)) { |
| 72 | info = -7; |
| 73 | } |
| 74 | |
| 75 | if (info != 0) { |
| 76 | return; // info; |
| 77 | } |
| 78 | |
| 79 | if (m == 0 || n == 0) { return; } |
| 80 | |
| 81 | switch (uplo) { |
| 82 | case MagmaFull: |
| 83 | return laset<T, 0>(m, n, offdiag, diag, dA, dA_offset, ldda, queue); |
| 84 | case MagmaLower: |
| 85 | return laset<T, 1>(m, n, offdiag, diag, dA, dA_offset, ldda, queue); |
| 86 | case MagmaUpper: |
| 87 | return laset<T, 2>(m, n, offdiag, diag, dA, dA_offset, ldda, queue); |
| 88 | default: return; |
| 89 | } |
| 90 | } |
| 91 | |
| 92 | #define INSTANTIATE(T) \ |
| 93 | template void magmablas_laset<T>( \ |