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]
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]
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]
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]
