45 lines
1.6 KiB
Factor
45 lines
1.6 KiB
Factor
! Copyright (C) 2010 Erik Charlebois.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: multiline locals io.encodings.ascii io.encodings.string sequences
|
|
math specialized-arrays alien.c-types math.order alien opencl tools.test
|
|
accessors arrays destructors kernel namespaces ;
|
|
FROM: alien.c-types => float ;
|
|
SPECIALIZED-ARRAY: float
|
|
IN: opencl.tests
|
|
|
|
STRING: kernel-source
|
|
__kernel void square(
|
|
__global float* input,
|
|
__global float* output,
|
|
const unsigned int count)
|
|
{
|
|
int i = get_global_id(0);
|
|
if (i < count)
|
|
output[i] = input[i] * input[i];
|
|
}
|
|
;
|
|
|
|
:: opencl-square ( in -- out )
|
|
[
|
|
in byte-length :> num-bytes
|
|
in length :> num-floats
|
|
cl-platforms first devices>> first :> device
|
|
device 1array <cl-context> &dispose :> context
|
|
context device f f <cl-queue> &dispose :> queue
|
|
|
|
context device queue [
|
|
"" kernel-source 1array <cl-program> &dispose "square" <cl-kernel> &dispose :> kernel
|
|
cl-read-access num-bytes in <cl-buffer> &dispose :> in-buffer
|
|
cl-write-access num-bytes f <cl-buffer> &dispose :> out-buffer
|
|
|
|
kernel in-buffer out-buffer num-floats uint <ref> 3array
|
|
{ num-floats } [ ] cl-queue-kernel &dispose drop
|
|
|
|
cl-finish
|
|
out-buffer 0 num-bytes <cl-buffer-range> cl-read-buffer num-floats <direct-float-array>
|
|
] with-cl-state
|
|
] with-destructors ;
|
|
|
|
[ float-array{ 1.0 4.0 9.0 16.0 100.0 } ]
|
|
[ float-array{ 1.0 2.0 3.0 4.0 10.0 } opencl-square ] unit-test
|