5

これは、私を夢中にさせている単純なOpenCL行列乗算カーネルです。

ちなみに私はpyopenclを使っています。

__kernel void matrixMul(  __global int* C,
                          __global int* A,
                          __global int* B,
                          int wA, int wB){

                int row = get_global_id(1); //2D Threas ID x
                int col = get_global_id(0); //2D Threas ID y

                //Perform dot-product accumulated into value
                int value = 0;
                for ( int k = 0; k < wA; k++ ){
                    value += A[row*wA + k] * B[k*wB+col];
                }
                C[row*wA+col] = value; //Write to the device memory
            }

どこで(入力)

A = [72 45
     75 61]
B = [26 53 
     46 76]
wA = wB = 2

私が得ている出力:

時々私は得る:

C = [3942 0
     0 5472]

そうでなければ私は得る:

C = [3942 7236
     3312 5472]

しかし、出力は次のようになります。

C = [3942 7236
     4756 8611]

ここでどのような間違いを犯しているのかわかりません。私は一日中運がないまま過ごしました。

これで私を助けてください

完全な Python コードは次のとおりです。

import pyopencl as cl
import numpy as np
import os

ORDER = 2
LEN = ORDER*ORDER
ctx = cl.create_some_context()

commandQueue = cl.CommandQueue( ctx )

A = np.array((72, 45, 75, 61), dtype = np.int32)
B = np.array((26, 53, 46, 76), dtype = np.int32)
C = np.empty_like(A)

in_buf1 = cl.Buffer( ctx, cl.mem_flags.READ_ONLY | cl.mem_flags.COPY_HOST_PTR,
                 hostbuf = A )
in_buf2 = cl.Buffer( ctx, cl.mem_flags.READ_ONLY | cl.mem_flags.COPY_HOST_PTR,
                 hostbuf = B )
out_buf = cl.Buffer( ctx, cl.mem_flags.WRITE_ONLY, C.nbytes )

kernelSrc1 = """__kernel void
            matrixMul(  /*const int Mdim,
                        const int Ndim,
                        const int Pdim,*/
                        __global int* C,
                        __global int* A,
                        __global int* B,
                        int wA, int wB)
           {
                int row = get_global_id(1); //2D Threas ID x
                int col = get_global_id(0); //2D Threas ID y                

                //Perform dot-product accumulated into value
                int value = 0;
                for ( int k = 0; k < wA; k++ ){
                    value += A[row*wA + k] * B[k*wB+col];
                }
                C[row*wA+col] = value; //Write to the device memory
            }"""

program1 = cl.Program(ctx, kernelSrc1 ).build()
event1 = program1.matrixMul( commandQueue, (LEN, ), None,
                     out_buf, in_buf1, in_buf2, np.int32(ORDER), np.int32(ORDER));
event1.wait()

cl.enqueue_copy(commandQueue, C, out_buf)
print C

Python 2.7.x、pyopencl 2012.1、AMD APP SDK を使用しています

4

1 に答える 1

7

グローバル サイズ引数を正しく設定していません。カーネルでグローバル サイズの 2 つの次元を使用しているため、グローバル サイズを (ORDER,ORDER)に設定する必要があります。それに変更すると、次のようになります。

[3942 7236
 4756 8611]
于 2012-10-21T19:02:34.257 に答える