50 lines
1.8 KiB
Plaintext
50 lines
1.8 KiB
Plaintext
//******************************************************************************
|
|
//
|
|
// File: OuterProductGpu.cu
|
|
// Package: edu.rit.gpu.example
|
|
// Unit: OuterProductGpu kernel function
|
|
//
|
|
// This C/CUDA source file is copyright (C) 2014 by Alan Kaminsky. All rights
|
|
// reserved. For further information, contact the author, Alan Kaminsky, at
|
|
// ark@cs.rit.edu.
|
|
//
|
|
// This C/CUDA source file is part of the Parallel Java 2 Library ("PJ2"). PJ2
|
|
// is free software; you can redistribute it and/or modify it under the terms of
|
|
// the GNU General Public License as published by the Free Software Foundation;
|
|
// either version 3 of the License, or (at your option) any later version.
|
|
//
|
|
// PJ2 is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
// WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
// A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
//
|
|
// A copy of the GNU General Public License is provided in the file gpl.txt. You
|
|
// may also obtain a copy of the GNU General Public License on the World Wide
|
|
// Web at http://www.gnu.org/licenses/gpl.html.
|
|
//
|
|
//******************************************************************************
|
|
|
|
/**
|
|
* Device kernel to compute the outer product matrix of two vectors.
|
|
* <P>
|
|
* Called with a two-dimensional grid of two-dimensional blocks.
|
|
*
|
|
* @param a First vector (input).
|
|
* @param b Second vector (input).
|
|
* @param c Outer product matrix (output).
|
|
* @param N Vector length.
|
|
*
|
|
* @author Alan Kaminsky
|
|
* @version 13-Apr-2014
|
|
*/
|
|
extern "C" __global__ void outerProduct
|
|
(double *a,
|
|
double *b,
|
|
double **c,
|
|
int N)
|
|
{
|
|
int row = blockIdx.y*blockDim.y + threadIdx.y;
|
|
int col = blockIdx.x*blockDim.x + threadIdx.x;
|
|
if (row < N && col < N)
|
|
c[row][col] = a[row]*b[col];
|
|
}
|