zlaset.c 9.16 KB
Newer Older
1
/**
2 3
 *
 * @file zlaset.c
4
 *
Mathieu Faverge's avatar
Mathieu Faverge committed
5 6
 * @copyright 2009-2014 The University of Tennessee and The University of
 *                      Tennessee Research Foundation. All rights reserved.
7 8
 * @copyright 2012-2014 Bordeaux INP, CNRS (LaBRI UMR 5800), Inria,
 *                      Univ. Bordeaux. All rights reserved.
9
 *
Mathieu Faverge's avatar
Mathieu Faverge committed
10
 ***
11 12 13 14 15 16
 *
 *
 *  MORSE computational routines
 *  MORSE is a software package provided by Univ. of Tennessee,
 *  Univ. of California Berkeley and Univ. of Colorado Denver
 *
Mathieu Faverge's avatar
Mathieu Faverge committed
17
 * @version 1.0.0
18 19 20 21 22 23 24 25 26
 * @comment This file has been automatically generated
 *          from Plasma 2.5.0 for MORSE 1.0.0
 * @author Mathieu Faverge
 * @author Emmanuel Agullo
 * @author Cedric Castagnede
 * @date 2010-11-15
 * @precisions normal z -> s d c
 *
 **/
27
#include "control/common.h"
28

29 30
/**
 ********************************************************************************
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
 *
 * @ingroup MORSE_Complex64_t
 *
 *  MORSE_zlaset copies all or part of a two-dimensional matrix A to another
 *  matrix B
 *
 *******************************************************************************
 *
 * @param[in] uplo
 *          Specifies the part of the matrix A to be copied to B.
 *            = MorseUpperLower: All the matrix A
 *            = MorseUpper: Upper triangular part is set. The lower
 *            triangle is unchanged.
 *            = MorseLower: Lower triangular part is set. The upper
 *            triangle is unchange.
 *
 * @param[in] M
48
 *          The number of rows of the matrix A. M >= 0.
49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
 *
 * @param[in] N
 *          The number of columns of the matrix A. N >= 0.
 *
 * @param[in] alpha
 *          All the offdiagonal array elements are set to alpha.
 *
 * @param[in] beta
 *          All the diagonal array elements are set to beta.
 *
 * @param[in,out] A
 *          On entry, the m by n matrix A.
 *          On exit, A(i,j) = ALPHA, 1 <= i <= m, 1 <= j <= n, i.ne.j;
 *                   A(i,i) = BETA,  1 <= i <= min(m,n)
 *
 * @param[in] LDA
 *          The leading dimension of the array A. LDA >= max(1,M).
 *
 *******************************************************************************
 *
 * @sa MORSE_zlaset_Tile
 * @sa MORSE_zlaset_Tile_Async
 * @sa MORSE_claset
 * @sa MORSE_dlaset
 * @sa MORSE_slaset
 *
 ******************************************************************************/
76 77 78
int MORSE_zlaset( MORSE_enum uplo, int M, int N,
                  MORSE_Complex64_t alpha, MORSE_Complex64_t beta,
                  MORSE_Complex64_t *A, int LDA )
79 80 81 82 83 84
{
    int NB;
    int status;
    MORSE_context_t *morse;
    MORSE_sequence_t *sequence = NULL;
    MORSE_request_t request = MORSE_REQUEST_INITIALIZER;
85
    MORSE_desc_t descAl, descAt;
86 87 88 89 90 91 92

    morse = morse_context_self();
    if (morse == NULL) {
        morse_fatal_error("MORSE_zlaset", "MORSE not initialized");
        return MORSE_ERR_NOT_INITIALIZED;
    }
    /* Check input arguments */
93
    if ( (uplo != MorseUpperLower) &&
94 95 96 97 98 99 100 101 102 103 104 105 106
         (uplo != MorseUpper) &&
         (uplo != MorseLower) ) {
        morse_error("MORSE_zlaset", "illegal value of uplo");
        return -1;
    }
    if (M < 0) {
        morse_error("MORSE_zlaset", "illegal value of M");
        return -2;
    }
    if (N < 0) {
        morse_error("MORSE_zlaset", "illegal value of N");
        return -3;
    }
107
    if (LDA < chameleon_max(1, M)) {
108 109 110 111 112
        morse_error("MORSE_zlaset", "illegal value of LDA");
        return -5;
    }

    /* Quick return */
Mathieu Faverge's avatar
Mathieu Faverge committed
113
    if (chameleon_min(N, M) == 0)
114
        return (double)0.0;
115 116 117 118 119 120 121 122 123 124 125

    /* Tune NB depending on M, N & NRHS; Set NBNB */
    status = morse_tune(MORSE_FUNC_ZGEMM, M, N, 0);
    if (status != MORSE_SUCCESS) {
        morse_error("MORSE_zlaset", "morse_tune() failed");
        return status;
    }

    /* Set NT */
    NB   = MORSE_NB;

126
    morse_sequence_create( morse, &sequence );
127

128
    /* Submit the matrix conversion */
129
    morse_zlap2tile( morse, &descAl, &descAt, MorseDescInout, uplo,
130
                     A, NB, NB, LDA, N, M, N, sequence, &request );
131 132

    /* Call the tile interface */
133
    MORSE_zlaset_Tile_Async( uplo, alpha, beta, &descAt, sequence, &request );
134

135 136 137
    /* Submit the matrix conversion back */
    morse_ztile2lap( morse, &descAl, &descAt,
                     MorseDescInout, uplo, sequence, &request );
Mathieu Faverge's avatar
Mathieu Faverge committed
138

139
    morse_sequence_wait( morse, sequence );
Mathieu Faverge's avatar
Mathieu Faverge committed
140

Mathieu Faverge's avatar
Mathieu Faverge committed
141 142
    /* Cleanup the temporary data */
    morse_ztile2lap_cleanup( morse, &descAl, &descAt );
143

144
    morse_sequence_destroy( morse, sequence );
145 146 147
    return MORSE_SUCCESS;
}

148 149
/**
 ********************************************************************************
150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
 *
 * @ingroup MORSE_Complex64_t_Tile
 *
 *  MORSE_zlaset_Tile - Tile equivalent of MORSE_zlaset().
 *  Operates on matrices stored by tiles.
 *  All matrices are passed through descriptors.
 *  All dimensions are taken from the descriptors.
 *
 *******************************************************************************
 *
 * @param[in] uplo
 *          Specifies the part of the matrix A to be copied to B.
 *            = MorseUpperLower: All the matrix A
 *            = MorseUpper: Upper triangular part
 *            = MorseLower: Lower triangular part
 *
 * @param[in,out] A
 *          On entry, the m by n matrix A.
 *          On exit, A(i,j) = ALPHA, 1 <= i <= m, 1 <= j <= n, i.ne.j;
 *                   A(i,i) = BETA,  1 <= i <= min(m,n)
 *
 *******************************************************************************
 *
 * @return
 *          \retval MORSE_SUCCESS successful exit
 *
 *******************************************************************************
 *
 * @sa MORSE_zlaset
 * @sa MORSE_zlaset_Tile_Async
 * @sa MORSE_claset_Tile
 * @sa MORSE_dlaset_Tile
 * @sa MORSE_slaset_Tile
 *
 ******************************************************************************/
185 186 187
int MORSE_zlaset_Tile( MORSE_enum uplo,
                       MORSE_Complex64_t alpha, MORSE_Complex64_t beta,
                       MORSE_desc_t *A )
188 189 190 191
{
    MORSE_context_t *morse;
    MORSE_sequence_t *sequence = NULL;
    MORSE_request_t request = MORSE_REQUEST_INITIALIZER;
192
    int status;
193 194 195 196 197 198

    morse = morse_context_self();
    if (morse == NULL) {
        morse_fatal_error("MORSE_zlaset_Tile", "MORSE not initialized");
        return MORSE_ERR_NOT_INITIALIZED;
    }
199 200
    morse_sequence_create( morse, &sequence );

201
    MORSE_zlaset_Tile_Async( uplo, alpha, beta, A, sequence, &request );
Mathieu Faverge's avatar
Mathieu Faverge committed
202

203
    MORSE_Desc_Flush( A, sequence );
Mathieu Faverge's avatar
Mathieu Faverge committed
204

205 206 207 208
    morse_sequence_wait( morse, sequence );
    status = sequence->status;
    morse_sequence_destroy( morse, sequence );
    return status;
209 210
}

211 212
/**
 ********************************************************************************
213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
 *
 * @ingroup MORSE_Complex64_t_Tile_Async
 *
 *  MORSE_zlaset_Tile_Async - Non-blocking equivalent of MORSE_zlaset_Tile().
 *  May return before the computation is finished.
 *  Allows for pipelining of operations at runtime.
 *
 *******************************************************************************
 *
 * @param[in] sequence
 *          Identifies the sequence of function calls that this call belongs to
 *          (for completion checks and exception handling purposes).
 *
 * @param[out] request
 *          Identifies this function call (for exception handling purposes).
 *
 *******************************************************************************
 *
 * @sa MORSE_zlaset
 * @sa MORSE_zlaset_Tile
 * @sa MORSE_claset_Tile_Async
 * @sa MORSE_dlaset_Tile_Async
 * @sa MORSE_slaset_Tile_Async
 *
 ******************************************************************************/
238 239 240 241
int MORSE_zlaset_Tile_Async( MORSE_enum uplo,
                             MORSE_Complex64_t alpha, MORSE_Complex64_t beta,
                             MORSE_desc_t *A,
                             MORSE_sequence_t *sequence, MORSE_request_t *request )
242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
{
    MORSE_context_t *morse;

    morse = morse_context_self();
    if (morse == NULL) {
        morse_fatal_error("MORSE_zlaset_Tile_Async", "MORSE not initialized");
        return MORSE_ERR_NOT_INITIALIZED;
    }
    if (sequence == NULL) {
        morse_fatal_error("MORSE_zlaset_Tile_Async", "NULL sequence");
        return MORSE_ERR_UNALLOCATED;
    }
    if (request == NULL) {
        morse_fatal_error("MORSE_zlaset_Tile_Async", "NULL request");
        return MORSE_ERR_UNALLOCATED;
    }
    /* Check sequence status */
Mathieu Faverge's avatar
Mathieu Faverge committed
259
    if (sequence->status == MORSE_SUCCESS) {
260
        request->status = MORSE_SUCCESS;
Mathieu Faverge's avatar
Mathieu Faverge committed
261 262
    }
    else {
263
        return morse_request_fail(sequence, request, MORSE_ERR_SEQUENCE_FLUSHED);
Mathieu Faverge's avatar
Mathieu Faverge committed
264
    }
265 266 267 268 269 270 271 272 273 274 275 276

    /* Check descriptors for correctness */
    if (morse_desc_check(A) != MORSE_SUCCESS) {
        morse_error("MORSE_zlaset_Tile_Async", "invalid descriptor");
        return morse_request_fail(sequence, request, MORSE_ERR_ILLEGAL_VALUE);
    }
    /* Check input arguments */
    if (A->nb != A->mb) {
        morse_error("MORSE_zlaset_Tile_Async", "only square tiles supported");
        return morse_request_fail(sequence, request, MORSE_ERR_ILLEGAL_VALUE);
    }
    /* Check input arguments */
277
    if ( (uplo != MorseUpperLower) &&
278 279 280 281 282 283
         (uplo != MorseUpper) &&
         (uplo != MorseLower) ) {
        morse_error("MORSE_zlaset_Tile_Async", "illegal value of uplo");
        return -1;
    }
    /* Quick return */
Mathieu Faverge's avatar
Mathieu Faverge committed
284
    if (chameleon_min(A->m, A->n) == 0) {
285 286 287
        return MORSE_SUCCESS;
    }

Mathieu Faverge's avatar
Mathieu Faverge committed
288
    morse_pzlaset( uplo, alpha, beta, A, sequence, request );
289 290 291

    return MORSE_SUCCESS;
}