util/X86MAC64/cuda/samples/6_Advanced/matrixMulDynlinkJIT/matrixMul.h

   1 /*
   2  * Copyright 1993-2013 NVIDIA Corporation.  All rights reserved.
   3  *
   4  * Please refer to the NVIDIA end user license agreement (EULA) associated
   5  * with this source code for terms and conditions that govern your use of
   6  * this software. Any use, reproduction, disclosure, or distribution of
   7  * this software and related documentation outside the terms of the EULA
   8  * is strictly prohibited.
   9  *
  10  */
  11
  12
  13 #ifndef _MATRIXMUL_H_
  14 #define _MATRIXMUL_H_
  15
  16 // Matrix dimensions
  17 // (chosen as multiples of the thread block size for simplicity)
  18 #define WA (4 * block_size) // Matrix A width
  19 #define HA (6 * block_size) // Matrix A height
  20 #define WB (4 * block_size) // Matrix B width
  21 #define HB WA  // Matrix B height
  22 #define WC WB  // Matrix C width
  23 #define HC HA  // Matrix C height
  24
  25 #endif // _MATRIXMUL_H_