コード例 #1
0
ファイル: clcomp.py プロジェクト: TUD-UCB-Boda/tud_boda
def run_conv( prog, queue ):
    func = prog['conv__num_imgs_20__in_pad_3__in_dim_0_227__in_dim_1_227__conv_has_relu_1__kern_sz_7__stride_2__out_chans_64__in_chans_3']
    print func
    func.argtypes = (cl.cl_mem, cl.cl_mem, cl.cl_mem, cl.cl_mem)
    in_ar = cl.array('f', range(100 * 1000 * 1000))
    in_buf, in_evt = cl.buffer_from_pyarray(queue, in_ar, blocking=False)
    filt_buf = in_buf.empty_like_this()
    bias_buf = in_buf.empty_like_this()
    out_buf = in_buf.empty_like_this()
    #run_evt = func(filt_buf, bias_buf, in_buf, out_buf).on(queue, gsize=(63,), lsize=(32,), wait_for=in_evt)
    func.setarg( 0, filt_buf )
    func.setarg( 1, bias_buf )
    func.setarg( 2, in_buf )
    func.setarg( 3, out_buf )
    run_evt = cl.clEnqueueNDRangeKernel( queue, func, gsize=(120*2166,), lsize=(120,), wait_for=in_evt)
    out, evt = cl.buffer_to_pyarray(queue, out_buf, wait_for=run_evt, like=in_ar)

    print "start wait"
    evt.wait()
    print "end wait, start loop"
    for i in range(1000):
        run_evt = cl.clEnqueueNDRangeKernel( queue, func, gsize=(120*2166,), lsize=(120,) )
        run_evt.wait()
    print "end loop"
    print out[0:10]
コード例 #2
0
ファイル: clcomp.py プロジェクト: pombredanne/boda
def run_conv(prog, queue):
    func = prog[
        "conv__num_imgs_20__in_pad_3__in_dim_0_227__in_dim_1_227__conv_has_relu_1__kern_sz_7__stride_2__out_chans_64__in_chans_3"
    ]
    print func
    func.argtypes = (cl.cl_mem, cl.cl_mem, cl.cl_mem, cl.cl_mem)
    in_ar = cl.array("f", range(100 * 1000 * 1000))
    in_buf, in_evt = cl.buffer_from_pyarray(queue, in_ar, blocking=False)
    filt_buf = in_buf.empty_like_this()
    bias_buf = in_buf.empty_like_this()
    out_buf = in_buf.empty_like_this()
    # run_evt = func(filt_buf, bias_buf, in_buf, out_buf).on(queue, gsize=(63,), lsize=(32,), wait_for=in_evt)
    func.setarg(0, filt_buf)
    func.setarg(1, bias_buf)
    func.setarg(2, in_buf)
    func.setarg(3, out_buf)
    run_evt = cl.clEnqueueNDRangeKernel(queue, func, gsize=(120 * 2166,), lsize=(120,), wait_for=in_evt)
    out, evt = cl.buffer_to_pyarray(queue, out_buf, wait_for=run_evt, like=in_ar)

    print "start wait"
    evt.wait()
    print "end wait, start loop"
    for i in range(1000):
        run_evt = cl.clEnqueueNDRangeKernel(queue, func, gsize=(120 * 2166,), lsize=(120,))
        run_evt.wait()
    print "end loop"
    print out[0:10]
コード例 #3
0
ファイル: clcomp.py プロジェクト: TUD-UCB-Boda/tud_boda
def run_mxplusb( prog, queue ):
    func = prog['mxplusb']
    print func
    func.argtypes = (cl.cl_float, cl.cl_mem, cl.cl_float, cl.cl_mem)
    x = cl.array('f', range(100))
    x_buf, in_evt = cl.buffer_from_pyarray(queue, x, blocking=False)
    y_buf = x_buf.empty_like_this()
    run_evt = func(2, x_buf, 5, y_buf).on(queue, len(x), wait_for=in_evt)
    y, evt = cl.buffer_to_pyarray(queue, y_buf, wait_for=run_evt, like=x)
    evt.wait()
    print y[0:10]
コード例 #4
0
ファイル: clcomp.py プロジェクト: pombredanne/boda
def run_mxplusb(prog, queue):
    func = prog["mxplusb"]
    print func
    func.argtypes = (cl.cl_float, cl.cl_mem, cl.cl_float, cl.cl_mem)
    x = cl.array("f", range(100))
    x_buf, in_evt = cl.buffer_from_pyarray(queue, x, blocking=False)
    y_buf = x_buf.empty_like_this()
    run_evt = func(2, x_buf, 5, y_buf).on(queue, len(x), wait_for=in_evt)
    y, evt = cl.buffer_to_pyarray(queue, y_buf, wait_for=run_evt, like=x)
    evt.wait()
    print y[0:10]