Skip to content
Snippets Groups Projects
Commit a423143e authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Merge branch 'master' of t:src/pyopencl

parents 96de8bd9 0ec05353
No related branches found
No related tags found
No related merge requests found
......@@ -161,8 +161,12 @@ h_c = numpy.empty((c_height, c_width)).astype(numpy.float32)
kernel_params = {"block_size": block_size,
"w_a":a_width, "h_a":a_height, "w_b":b_width}
if "NVIDIA" in queue.device.vendor:
options = "-cl-mad-enable -cl-fast-relaxed-math"
else:
options = None
prg = cl.Program(ctx, KERNEL_CODE % kernel_params,
).build(options="-cl-mad-enable -cl-fast-relaxed-math")
).build(options=options)
kernel = prg.matrixMul
#print prg.binaries[0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment