sgemm.cu

Nividia提供的CUDA的BLAS库源码

/*
 * Copyright 1993-2008 NVIDIA Corporation.  All rights reserved.
 *
 * NOTICE TO USER:   
 *
 * This source code is subject to NVIDIA ownership rights under U.S. and
 * international Copyright laws.  
 *
 * This software and the information contained herein is being provided 
 * under the terms and conditions of a Source Code License Agreement.     
 *
 * NVIDIA MAKES NO REPRESENTATION ABOUT THE SUITABILITY OF THIS SOURCE
 * CODE FOR ANY PURPOSE.  IT IS PROVIDED "AS IS" WITHOUT EXPRESS OR 
 * IMPLIED WARRANTY OF ANY KIND.  NVIDIA DISCLAIMS ALL WARRANTIES WITH
 * REGARD TO THIS SOURCE CODE, INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY, NONINFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE.
 * IN NO EVENT SHALL NVIDIA BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL,
 * OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
 * OF USE, DATA OR PROFITS,  WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
 * OR OTHER TORTIOUS ACTION,  ARISING OUT OF OR IN CONNECTION WITH THE USE
 * OR PERFORMANCE OF THIS SOURCE CODE.  
 *
 * U.S. Government End Users.   This source code is a "commercial item" as 
 * that term is defined at  48 C.F.R. 2.101 (OCT 1995), consisting  of
 * "commercial computer  software"  and "commercial computer software 
 * documentation" as such terms are  used in 48 C.F.R. 12.212 (SEPT 1995)
 * and is provided to the U.S. Government only as a commercial end item.
 * Consistent with 48 C.F.R.12.212 and 48 C.F.R. 227.7202-1 through
 * 227.7202-4 (JUNE 1995), all U.S. Government End Users acquire the 
 * source code with only those rights set forth herein.
 */

/* Use square 32x32 tiles to access and cache portions of source matrices A,B 
 * and result matrix C
 */
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include "cublasP.h"
#define TILE_DIM_LOG    (5)
#define THREAD_COUNT    (CUBLAS_SGEMM_LARGE_THREAD_COUNT)
#include "sgemm_sizing.h"
#include "sgemm_common.h"

texture<float> texA;
texture<float> texB;

__shared__ float AA[(TILE_DIM+1)*TILE_DIM]; /*pad to elim. GRF bank conflicts*/
__shared__ float BB[(TILE_DIM+1)*TILE_DIM]; /*pad to elim. GRF bank conflicts*/

typedef void (*pf) (struct cublasSgemmParams parms);
static pf sgemm_sw[16] = {
    sgemm_main_gld_sw_na_nb,
    sgemm_main_gld_sw_na_tb,
    sgemm_main_gld_sw_ta_nb,
    sgemm_main_gld_sw_ta_tb,
    sgemm_main_gld_sw_na_nb_fulltile,
    sgemm_main_gld_sw_na_tb_fulltile,
    sgemm_main_gld_sw_ta_nb_fulltile,
    sgemm_main_gld_sw_ta_tb_fulltile,
    sgemm_main_tex_sw_na_nb,
    sgemm_main_tex_sw_na_tb,
    sgemm_main_tex_sw_ta_nb,
    sgemm_main_tex_sw_ta_tb,
    sgemm_main_tex_sw_na_nb_fulltile,
    sgemm_main_tex_sw_na_tb_fulltile,
    sgemm_main_tex_sw_ta_nb_fulltile,
    sgemm_main_tex_sw_ta_tb_fulltile
};

static pf sgemm_hw[16] = {
    sgemm_main_gld_hw_na_nb,
    sgemm_main_gld_hw_na_tb,
    sgemm_main_gld_hw_ta_nb,
    sgemm_main_gld_hw_ta_tb,
    sgemm_main_gld_hw_na_nb_fulltile,
    sgemm_main_gld_hw_na_tb_fulltile,
    sgemm_main_gld_hw_ta_nb_fulltile,
    sgemm_main_gld_hw_ta_tb_fulltile,
    sgemm_main_tex_hw_na_nb,
    sgemm_main_tex_hw_na_tb,
    sgemm_main_tex_hw_ta_nb,
    sgemm_main_tex_hw_ta_tb,
    sgemm_main_tex_hw_na_nb_fulltile,
    sgemm_main_tex_hw_na_tb_fulltile,
    sgemm_main_tex_hw_ta_nb_fulltile,
    sgemm_main_tex_hw_ta_tb_fulltile
};

__host__ void cublasLargeSgemm (struct cublasContext *ctx, char transa, 
                                char transb, int m, int n, int k, 
                                float alpha, const float *A, int lda, 
                                const float *B, int ldb, float beta, float *C, 
                                int ldc)
{
    struct cublasSgemmParams params;
    cudaError_t cudaStat;
    int fullTilesOnly;
    size_t texAOfs = 0;
    size_t texBOfs = 0;
    int sizeA = lda * ((toupper(transa) == 'N') ? k : m);
    int sizeB = ldb * ((toupper(transb) == 'N') ? n : k);
    int useTexture;
    int usePureHwStepper;
    int funcIdx;
    dim3 ctaDimsHw (((n+TILE_DIM-1)/TILE_DIM), ((m+TILE_DIM-1)/TILE_DIM));
    dim3 ctaDimsSw (CUBLAS_SGEMM_GRIDW, CUBLAS_SGEMM_GRIDH);

    if (!cublasInitialized (ctx)) {
        cublasSetError (ctx, CUBLAS_STATUS_NOT_INITIALIZED);
        return;
    }

    /* We can only use texture if the matrices fit into the largest matrix 
     * size supported.
     */
    useTexture = ((sizeA < CUBLAS_MAX_1DBUF_SIZE) &&
                  (sizeB < CUBLAS_MAX_1DBUF_SIZE));

    /* choose HW-only stepping if dimensions of result matrix do not exceed the
     * maximum CTA grid dimensions.
     */
    usePureHwStepper = ((m < (CUBLAS_CTA_MAX_DIM * TILE_DIM)) &&
                        (n < (CUBLAS_CTA_MAX_DIM * TILE_DIM)));

    /* we can eliminate checking for endcases if we know all tiles are fully
     * populated. Important benchmark case!
     */
    fullTilesOnly = (((m % TILE_DIM) == 0) &&
                     ((n % TILE_DIM) == 0) &&
                     ((k % TILE_DIM) == 0));


    /* For matrices where each row is aligned, GLD coalesces nicely and is
     * faster, so don't use texture.
     */
    if (!(((ptrdiff_t)A)%CUBLAS_WORD_ALIGN) && 
        !(((ptrdiff_t)B)%CUBLAS_WORD_ALIGN) &&
        !(lda%(CUBLAS_WORD_ALIGN/sizeof(A[0]))) &&
        !(ldb%(CUBLAS_WORD_ALIGN/sizeof(B[0])))) {
        useTexture = 0;
    }

    if (useTexture){
        if ((cudaStat=cudaBindTexture (&texAOfs,texA,A,sizeA*sizeof(A[0]))) != cudaSuccess) {
            cublasSetError (ctx, CUBLAS_STATUS_MAPPING_ERROR);
            return;
        }
        if ((cudaStat=cudaBindTexture (&texBOfs,texB,B,sizeB*sizeof(B[0]))) != cudaSuccess) {
            cudaUnbindTexture (texA);
            cublasSetError (ctx, CUBLAS_STATUS_MAPPING_ERROR);
            return;
        }
        texAOfs /= sizeof(A[0]);
        texBOfs /= sizeof(B[0]);
    }

    memset (&params, 0, sizeof(params));
    params.transa = ((toupper(transa) == 'T') || (toupper(transa) == 'C'));
    params.transb = ((toupper(transb) == 'T') || (toupper(transb) == 'C'));
    params.m = m;
    params.n = n;
    params.k = k;
    params.alpha = alpha;
    params.A = A;
    params.lda = lda;
    params.B = B;
    params.ldb = ldb;
    params.beta = beta;
    params.C = C;
    params.ldc =ldc;
    params.texAOfs = (int)texAOfs;
    params.texBOfs = (int)texBOfs;

    funcIdx = ((useTexture << 3) | (fullTilesOnly << 2) | 
               (params.transa << 1) | params.transb);

    cudaStat = cudaGetLastError(); /* clear error status */
    if (usePureHwStepper) {
        sgemm_hw[funcIdx]<<<ctaDimsHw,THREAD_COUNT>>>(params);
    } else {
        sgemm_sw[funcIdx]<<<ctaDimsSw,THREAD_COUNT>>>(params);
    }
    cudaStat = cudaGetLastError(); /* check for launch error */

    if (cudaStat != cudaSuccess) {
        cublasSetError (ctx, CUBLAS_STATUS_EXECUTION_FAILED);
    }
    if (useTexture) {
        if ((cudaStat = cudaUnbindTexture (texA)) != cudaSuccess) {
            cublasSetError (ctx, CUBLAS_STATUS_INTERNAL_ERROR);
        }
        if ((cudaStat = cudaUnbindTexture (texB)) != cudaSuccess) {
            cublasSetError (ctx, CUBLAS_STATUS_INTERNAL_ERROR);
        }
    }
}

__global__ void sgemm_main_gld_hw_na_nb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 0
#define TRANSA            0
#define TRANSB            0
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_na_tb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 0
#define TRANSA            0
#define TRANSB            1
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_ta_nb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 0
#define TRANSA            1
#define TRANSB            0
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_ta_tb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY   
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 0
#define TRANSA            1
#define TRANSB            1
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_na_nb_fulltile (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   1
#define USE_MIXED_STEPPER 0
#define TRANSA            0
#define TRANSB            0
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_na_tb_fulltile (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   1
#define USE_MIXED_STEPPER 0
#define TRANSA            0
#define TRANSB            1
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_ta_nb_fulltile (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   1
#define USE_MIXED_STEPPER 0
#define TRANSA            1
#define TRANSB            0
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_hw_ta_tb_fulltile (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY   
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   1
#define USE_MIXED_STEPPER 0
#define TRANSA            1
#define TRANSB            1
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_sw_na_nb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY 
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 1
#define TRANSA            0
#define TRANSB            0
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_sw_na_tb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY
#undef  USE_MIXED_STEPPER
#undef  TRANSA
#undef  TRANSB
#undef  FAST_IMUL
#undef  USE_TEX
#define FULL_TILES_ONLY   0
#define USE_MIXED_STEPPER 1
#define TRANSA            0
#define TRANSB            1
#define FAST_IMUL         0
#define USE_TEX           0
#include "sgemm.h"
}

__global__ void sgemm_main_gld_sw_ta_nb (struct cublasSgemmParams parms)
{
#undef  FULL_TILES_ONLY