第12章 WebCL:使用OpenCL加速Web应用 - 12.2 如何使用WebCL编程

优质
小牛编辑
127浏览
2023-12-01

WebCL 1.0其实就是使用JavaScript实现的OpenCL 1.2。同样的OpenCL API、语法以及运行时,无需太多行代码就能完成(JavaScript是面向对象的语言)。我们通常会将WebCL与其他类似的技术联系在一起进行比较。如果你对WebGL很不了解,也没有关系。

与OpenCL一样,WebCL的编程也是由两部分构成:

  1. 主机端(例如,Web浏览器)用来控制和执行JavaScript程序
  2. 设备端(例如,GPU)用来进行计算——OpenCL内核

和WebGL一样,WebCL也属于窗口对象的特性。首先,我们要检查WebCL是否可用,如果可用,则创建对应的上下文对象:

  1. // First check if the WebCL extension is installed at all
  2. if (window.webcl == undefined){
  3. alert("Unfortunately your system does not support WebCL." +
  4. "Make sure that you have both the OpenCL dirver" +
  5. "and the WebCL browser extension installed.");
  6. }
  7. // Get a list of available CL platforms, and another list of the
  8. // available devices on each platform. If there are no platforms
  9. // or no available devices on any platform, then we can conclude
  10. // that WebCL is not available
  11. webcl = window.webcl
  12. try{
  13. var platforms = webcl.getPlatforms();
  14. var devices = [];
  15. for (var i in platforms){
  16. var p = platforms[i];
  17. devices[i] = p.getDevices();
  18. }
  19. alert("Excellent! Your system does support WebCL");
  20. } catch(e){
  21. alert("Unfortunately platform or device inquiry failed.");
  22. }
  23. // Setup WebCL context using the default device
  24. var ctx = webcl.createContext();

12.2 如何使用WebCL编程 - 图1

图12.1 WebCL对象

图12.1中描述了WebCL中的对象结构关系。注意WebCL是接口的名字,而webcl是JavaScript的对象。更多信息可以查阅WebCL的标准[1]。

Web应用无法获取平台信息之后,列出所有可以用的设备;而是通过获取到的设备信息了解对应平台,从而指定对应的设备或平台。

  1. // Find appropriate device
  2. for (var j = 0, jl = device.length; j < jl; ++j){
  3. var d = devices[j];
  4. var devExts = d.getInfo(cl.DEVICE_EXTENSIONS);
  5. var devGMem = d.getInfo(cl.DEVICE_GLOBAL_MEM_SIZE);
  6. var devLMem = d.getInfo(cl.DEVICE_LOCAL_MEM_SIZE);
  7. var devCompUnits = d.getInfo(cl.DEVICE_MAX_COMPUTE_UNITS);
  8. var devHasImage = d.getInfo(cl.DEVICE_IMAGE_SUPPORT);
  9. // select device that matches your requirements
  10. platform = ...
  11. device = ...
  12. }
  13. // assuming we found the best device, we can create the context
  14. var context = webcl.createContext(platform, device);

应用会在运行时对OpenCL对象进行管理,比如:命令队列、内存对象、程序对象、内核对象,以及入队命令(比如:执行内核、读取内存对象或写入内存对象)。

WebCL中定义了如下对象:

  • 命令队列
  • 内存对象(数组图像)
  • 采样器对象,其描述了在内核中如何对图像进行读取
  • 程序对象,其包含了一些列内核函数
  • 内核对象,在内核源码中使用__kernel声明的函数,其作为真正的执行对象
  • 事件对象,其用来追踪命令执行状态,以及对一个命令进行性能分析
  • 命令同步对象,比如标记栅栏

首先我们需要创建程序对象。WebCL与WebGL 1.0类似,假设可以提供一段内核源码。这样的话,Web应用就需要内嵌一个编译器。源码先从设备上进行载入,之后进行编译。和其他编译器一样,OpenCL编译器也定义了一些编译选项。

  1. // Create the compute program from the source strings
  2. program = context.createProgram(source);
  3. // Build the program executable with relaxed math flag
  4. try{
  5. program.build(device, "-cl-fast-relaxed-math");
  6. } catch(err) {
  7. throw 'Error building program:' + err + program.getBuildInfo(device, cl.PROGRAM_BUILD_LOG);
  8. }

如此这般,我们的程序就可以编译了,并且一个程序对象内具有一个或者多个内核函数。这些内核函数作为我们程序的入口函数,就如同静态库中的接口一样。为了区别不同的内核函数,我们需要创建WebCLKernel对象:

  1. // Create the compute kernels from within the program
  2. var kernel = program.createKernel("kernel_function_name");

就像普通函数一样,内核函数通常都会有一些参数。JavaScript会提供一些数据类型,typed arrays[2]就是用来传递不同类型的内核参数(具体参数类型见表12.1)。对于其他类型的数据,我们可以使用WebCL对象:

  • WebCLBuffer和WebCLImage,可以用来包装数组
  • WebCLSampler可以对图像进行采样

一个WebCLBuffer对象可以将数据以一维的方式存储起来。数组中错存储的元素类型都可以使标量类型(比如:int或float),向量类型,或是自定义结构体类型。

表12.1 setArg()中使用的webcl.type与C类型之间的关系

内核参数类型setArg()值的类型setArg()数组类型注意
char, ucharscalarUint8Array, Int8Arrary1 byte
short, ushortscalarUint16Array, Int16Array2 bytes
int, uintscalarUint32Array, Int32Array4 bytes
long, ulongscalarUint64Array, Int64Array8 bytes
floatscalarFloat32Array4 bytes
charNvectorInt8Array for (u)charNN = 2,3,4,8,16
shortNvectorInt16Array for (u)shortNN = 2,3,4,8,16
intNvectorInt32Array for (u)intNN = 2,3,4,8,16
floatNvectorFloat32Array for floatN and halfNN = 2,3,4,8,16
doubleNvectorFloat64Array for (u)doubleNN = 2,3,4,8,16
char, …, double *WebCLBuffer
image2d_tWebCLImage
sampler_tWebCLSampler
__localInt32Array([size_in_bytes])内核内部定义大小
  1. // Create a 1D buffer
  2. var buffer = webcl.createBuffer(flags, sizeInBytes, optional srcBuffer);
  3. // flags:
  4. // webcl.MEM_READ_WRITE Default. Memory object is read and written by kernel
  5. // webcl.MEM_WRITE_ONLY Memory object only writeten by kernel
  6. // webcl.MEM_READ_ONLY Memory object only read by kernel
  7. // webcl.MEM_USE_HOST_PTR Implementation requests OpenCL to allocate host memory
  8. // webcl.MEM_COPY_HOST_PTR Implementation requests OpenCL to allocate host memory and copy data from srcBuffer memory. srcBuffer must be specified

注意,只有从内存对象或其子对象中读取数据,或读取多个具有重叠区域的子内存对象是有定义的。其他方式的并发读写都会产生未定义的行为。

WebCL图像对象可以存储1,2,3维的纹理,渲染内存或图像。图像对象中的元素可以从预定义的图像格式列表中选择。当前的WebCL版本最多只支持到二维图像。

  1. // create a 32-bit RGBA WebCLImage object
  2. // first, we define the format of the image
  3. var imageFormat = {
  4. // memory layout in which pixel data channels are stored in the image
  5. 'channelOrder':webcl.RGBA,
  6. // type of the channel data
  7. 'channelType':webcl.UNSIGNED_INT8,
  8. // image size
  9. 'width':image_width,
  10. 'height':image_height,
  11. // scan-line pitch in bytes.
  12. // If imageBuffer is null, which is the default if rowPitch is not specified.
  13. 'rowPitch':image_pitch
  14. };
  15. // Image on device
  16. // imageBuffer is a typed array that contain the image data already allocated by the application
  17. // imageBuffer.byteLength >= rowPitch * image_height. The size of each element in bytes must be a power of 2.
  18. var image = context.createImage(webcl.MEM_READ_ONLY | webcl.MEM_USE_HOST_PTR, imageFormat, imageBuffer);

WebCLSampler告诉内核函数读和对图像数据进行读取。WebCL的采样器和WebGL的采样器类似。

  1. // create a smpler object
  2. var sampler = context.createSampler(normalizedCoords, addressingMode, filterMode);
  3. // normalizedCoods indicates if image coordinates specified are normalized.
  4. // addressingMode indicated how out-of-range image coordinations are handled when reading an image.
  5. // This can be set to webcl.ADDRESS_MIRRORED_REPEAT
  6. // webcl.ADDRESS_REPEAT, webcl.ADDRESS_CLAMP_TO_EDGE,
  7. // webcl.ADDRESS_CLAMP and webcl.ADDRESS_NONE.
  8. // filterMode specifies the type of filter to apply when reading an image. This can be webcl.FILTER_NEAREST or webcl.FILTER_LINEAR

使用WebCLKernel.setArg()将标量、向量或内存对象以参数的形式传入内核。需要传递局部内存时,我们可以使用长度为1的Int32Array,其中的数值是需要分配多少字节的局部内存,因为局部内存不能在主机端或设备端初始化,不过主机端可以通过内核参数的形式告诉设备端要分配多少局部内存。

根据经验内核所需要的所有参数,都需要是一个对象。即使是标量也也要包装在长度为1的数组中。向量根据长度放置在数组中。数组、图像和采样器都是WebCL对象。内存对象(数组和图像)需要在内核执行命令入队前,通过主机内存转移到设备内存。

这里展示一个例子:

  1. // Sets value of kernel argument idx with value as memory object or sampler
  2. kernel.setArg(idx.a_buffer);
  3. kernel.setArg(idx.a_image);
  4. kernel.setArg(idx.a_sampler);
  5. // Sets value of argument 0 to the integer value 5
  6. kernel.setArg(0, new Int32Array([5]));
  7. // Sets value of argument 1 to the float value 1.34
  8. kernel.setArg(1, new Float32Array([1.34]));
  9. // Sets value of argument 2 as a 3-float vector
  10. // buffer should be a Float32Array with 3 floats
  11. kernel.setArg(2, new Float32Array([1.0, 2.0, 3.0]));
  12. // Allocate 4096 bytes of local memory for argument 4
  13. kernel.setArg(3, new Int32Array([4096]));

当需要传递一个标量时,需要告诉程序内核标量数据的类型。JavaScript只有一种类型——数字——所以我们需要通过setArg提供数据类型信息。

注意:

  • 长整型是64位整数,其无法在JavaScript中表示。其只能表示成两个32位整数:低32位存储在数组的第一个元素中,高32位存储在第二个元素中。
  • 如果使用__constant对内核参数进行修饰,那么其大小就不能超多webcl.DEVICE_MAX_CONSTANT_BUFFER_SIZE。
  • OpenCL允许通过数组传递自定义结构体,不过为了可移植性WebCL还不支持自定义结构体的传递。其中很重要的原因是因为主机端和设备端的存储模式可能不同(大端或小端),其还需要开发者对于不同端的设备进行数据整理,即使主机和设备位于同一设备上。
  • 所有WebCL API都是线程安全的,除了kernel.setArg()。不过,kernel.setArg()在被不同的内核对象并发访问的时候也是安全的。未定义的行为会发生在多个线程调用同一个WebCLKernel对象时。

WebCL的对象中,命令队列中包含一系列操作和命令。应用可能使用多个独立的命令队列,直到其中有数据共享时才进行同步。

命令需要有序入队,不过在设备端可能是顺序执行(默认),或者是乱序执行。乱序意味着当命令队列中包含两个命令A和B时,顺序的执行能保证A一定能在B之前执行,而乱序的话就不能保证A和B结束的顺序。对于乱序队列,可以使用等待事件或入队一个栅栏命令,来保证命令的之间的依赖顺序。乱序队列属于进阶的主题,本章就不再进行描述。对其感兴趣的读者可以去这里[3]看看。目前很多的OpenCL实现并不支持乱序队列。你可以尝试使用QUEUE_OUT_OF_ORDER_EXEC_MODE_ENABLE测试一下OpenCL实现是否支持乱序队列。如果得到的是INVALID_QUEUE_PROERTIES异常抛出的话,那么你所使用的设备就不支持乱序队列。

  1. // Create an in-order command-queue(default)
  2. var queue = context.createCommandQueue(device);
  3. // Create an in-order command-queue with profiling of commands enabled
  4. var queue = context.createCommandQueue(device, webcl.QUEUE_PROFILING_ENABLE);
  5. // Create an out-of-order command-queue
  6. var queue = context.createCommandQueue(device, webcl.QUEUE_OUT_OF_ORDER_EXEC_MODE_ENABLE);

命令队列在指定设备上创建。多个命令队列具有其相对应的设备。应用可以通过主机端和设备端的数据传输,重叠的执行内核。图12.2展示了将问题分成两部分可能会更快的完成:

12.2 如何使用WebCL编程 - 图2

  • 第一部分是主机端向设备端传输数据,分解之后只需要原来拷贝时间的一半即可完成工作。然后,执行内核,基本上也是一半的时间完成。最后将数据传回主机,同样还是原先一半的时间。
  • 第一个传输数据完成后,第二个才开始,有些类似于CPU的流水线。

当有一系列命令入队后,使用WebCL中的enqueueNDRange(kenrel, offsets, globals, locals)就能执行对应的命令:

  • kernel——执行命令的内核对象。
  • offsets——对全局区域的偏移。如果传null,则代表offsets=[0, 0, 0]。
  • globals——内核所要解决问题的尺寸。
  • locals——每个维度上的工作组中,工作项的数量。如果传null,设备会对其自行设定。

例如,如果我们要操作一个宽度width,高度为height大小的图像,那么globals可以设置为[width, height],locals是可以设置为[16, 16]。这里需要注意的是,如果enqueuNDRange()中locals的大小超过了webcl.KERNEL_WORK_GROUP_SIZE,enqueuNDRange()就会执行失败。


[1] WebCL 1.0 Specification, http://www.khronos.org/registry/webcl/specs/latest/1.0

[2] Typed Array Specification, http://www.khronos.org/registry/typedarray/specs/latest

[3] Derek Gerstmann Siggraph Asia 2009 on Advanced OpenCL Event Model Usage, http://sa09.ida.ucdavis.edu/docs/SA09-opencl-dg-events-stream.pdf