Advertisement
Guest User

Untitled

a guest
Apr 23rd, 2019
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.33 KB | None | 0 0
  1. __kernel void matrix_mult(__global int * a, __global int * b, __global int * c, int n)
  2. {
  3.    int row = get_global_id(0);
  4.    int col = get_global_id(1);
  5.  
  6.    if (row >= n || col >= n)
  7.       return;
  8.  
  9.    int sum = 0;
  10.  
  11.    for (int k = 0; k < n; ++k)
  12.       sum += a[row * n + k] * b[k * n + col];
  13.  
  14.    c[row * n + col] = sum;
  15. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement