You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
370 B
14 lines
370 B
1 month ago
|
from tinygrad.runtime.ops_gpu import CLDevice, CLProgram, compile_cl
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
dev = CLDevice()
|
||
|
|
lib = compile_cl("""
|
||
|
|
#pragma OPENCL EXTENSION cl_khr_fp16 : enable
|
||
|
|
__kernel void test(__global half *out, __global half *a, __global half *b) {
|
||
|
|
int gid = get_global_id(0);
|
||
|
|
out[gid] = max(a[gid], b[gid]);
|
||
|
|
}
|
||
|
|
""")
|
||
|
|
prg = CLProgram(dev, "test", lib)
|
||
|
|
|