Python/OpenCL
< Python
Jump to navigation
Jump to search
PyOpenCL
Type Mapping
| OpenCL | Python Input | Python Output |
|---|---|---|
| float | np.float32 | np.float32 |
| double | np.float64 | np.float64 |
| int | np.int32 | np.int32 |
| long | np.int64 | np.int64 |
| char | b'...' | bytearray |
See Also:
Sample
import os
import numpy as np
import pyopencl as cl
# Kernel code.
CL_INC = '''
__kernel void inc_f32(__global const float *a_g, __global float *res_g)
{
int gid = get_global_id(0);
res_g[gid] = a_g[gid] + 1.0;
}
__kernel void inc_f64(__global const double *a_g, __global double *res_g)
{
int gid = get_global_id(0);
res_g[gid] = a_g[gid] + 1.0;
}
__kernel void inc_i32(__global const int *a_g, __global int *res_g)
{
int gid = get_global_id(0);
res_g[gid] = a_g[gid] + 1;
}
__kernel void inc_i64(__global const long *a_g, __global long *res_g)
{
int gid = get_global_id(0);
res_g[gid] = a_g[gid] + 1;
}
'''
# Build
os.environ['PYOPENCL_CTX'] = '0:1' # Apple > HD Graphics 4000 of Macbook Air 2012 mid
ctx = cl.create_some_context()
queue = cl.CommandQueue(ctx)
mf = cl.mem_flags
prg = cl.Program(ctx, CL_INC).build()
# Shorthand to call kernel function.
def call_kernel(kernel_name, *args):
in_np = args[0]
out_np = np.empty_like(in_np)
in_cl = cl.Buffer(ctx, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=in_np)
out_cl = cl.Buffer(ctx, mf.WRITE_ONLY, in_np.nbytes)
getattr(prg, kernel_name)(queue, in_np.shape, None, in_cl, out_cl)
cl.enqueue_copy(queue, out_np, out_cl)
return out_np
# Test inc_f32()
in_np = np.array([1.0, 2.0, 3.0, 4.0, 5.0]).astype(np.float32)
out_np = call_kernel('inc_f32', in_np)
print('Results of inc_f32()')
print(in_np)
print(out_np)
print()
# Test inc_f64()
in_np = np.array([1.0, 2.0, 3.0, 4.0, 5.0])
out_np = call_kernel('inc_f64', in_np)
print('Results of inc_f64()')
print(in_np)
print(out_np)
print()
# Test inc_i32()
in_np = np.array([1, 2, 3, 4, 5]).astype(np.int32)
out_np = call_kernel('inc_i32', in_np)
print('Results of inc_i32()')
print(in_np)
print(out_np)
print()
# Test inc_i64()
in_np = np.array([1, 2, 3, 4, 5])
out_np = call_kernel('inc_i64', in_np)
print('Results of inc_i64()')
print(in_np)
print(out_np)
print()
numpy
Data Types
import numpy as np
# Default int type is int64
a = np.array([1, 2, 3])
print(type(a[0]).__name__)
# Default float type is float64
b = np.array([1.0, 2.0, 3.0])
print(type(b[0]).__name__)
# Force int32
c = np.array([1, 2, 3]).astype(np.int32)
print(type(c[0]).__name__)
# Force float32
d = np.array([1.0, 2.0, 3.0]).astype(np.float32)
print(type(d[0]).__name__)
Merge Sort
Sort 8 items
Sort 5 items
Fontconfig error: No writable cache directories
Fontconfig error: No writable cache directories Fontconfig error: No writable cache directories Fontconfig error: No writable cache directories failed at node 258[1] dot: maze.c:313: chkSgraph: Assertion `np->cells[1]' failed. Aborted (core dumped)
digraph s10 {
// options
// theme = warm
// usage =
// default settings of graphs
graph [
rankdir = LR,
color = "#804000",
bgcolor = "#fffff7",
fontcolor = "#000000",
fontsize = 12,
style = dashed,
gradientangle = 65,
splines = ortho,
];
// default settings of nodes
node [
shape = box,
style = "filled,rounded",
height = 0.3,
fontsize = 10,
// theme
color = "#c07000",
fontcolor = "#000000",
fillcolor = "#ffffff:#ffffc0",
gradientangle = 295 // left, top -> right, bottom
];
// default settings of edges
edge [
color = "#704000",
fontcolor = "#704000",
fontsize = 10,
arrowsize = 0.6
];
// nodes, edges, and clusters
rankdir="TB";
A [label="5 4 3 2 1"];
subgraph cluster_1st {
B1 [label="5 | 4"];
B2 [label="3 | 2"];
B3 [label="1"];
C1 [label="4 5"];
C2 [label="2 3"];
C3 [label="1"];
}
D1 [label="4 5 | 2 3"];
D2 [label="1"];
E1 [label="2 3 4 5"];
E2 [label="1"];
F1 [label="2 3 4 5 | 1"];
G1 [label="1 2 3 4 5"];
A -> {B1 B2 B3};
B1 -> C1;
B2 -> C2;
B3 -> C3;
{C1 C2} -> D1;
{C3} -> D2;
D1 -> E1;
D2 -> E2;
{E1 E2} -> F1;
F1 -> G1;
}