为Halide安装opencl支持

最新推荐文章于 2024-10-11 07:57:36 发布

原创

最新推荐文章于 2024-10-11 07:57:36 发布 · 640 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#linux #开发语言 #c++ #人工智能 #算法

本文详细介绍了如何为Halide安装OpenCL支持，包括下载源码、编译和开启OpenCL选项。通过示例展示了在Halide中调用OpenCL的流程，包括计算图的构建、JIT编译和GPU处理。实验结果显示，启用OpenCL后，推理速度有所提升。解析了Halide调用OpenCL的过程，包括创建会话、编译程序、分配内存、设置内核参数和执行命令。

为Halide安装opencl支持

1、背景

上次我们从源码编译了Halide，进行了代码示例学习；同时，我们也在该专题下进行过Intel-Cpu-Opencl的安装教程，同样有着相关示例学习和工程源码。

2、为Halide安装Opencl支持的目的

主要目的就是想要看下是否可以使用Halide进行opencl核函数的实现，以及想要看下halide调度对核函数的是如何产生影响的。同时，想看下其对后续的推理框架进行核函数生成的帮助，以及自创AutoKernel工具的可行性。

3、安装opencl支持的过程

首先安装Intel-Cpu-OpenCL-Runtime-SDK，具体方便见过往文章链接
Halide源码编译开启opencl支持，具体见过往文章链接


# 1.下载Halide的源码
curl -o Halide.zip https://codeload.github.com/halide/Halide/zip/refs/heads/main
# 2.解压文件
unzip Halide.zip
# 3.编译
if [ ! -d "./build" ]; then
 	mkdir build
fi
cd build
# 关闭python_bindings开关
#在配置运行cmake的时候，打开target_opencl开关
cmake -DCMAKE_BUILD_TYPE=Release -DLLVM_DIR=$LLVM_ROOT/lib/cmake/llvm -DWITH_PYTHON_BINDINGS=OFF -DCMAKE_INSTALL_PREFIX=`pwd`/opencl-build -DTARGET_OPENCL=ON -S ..
cmake --build . -j32
# 4.此时安装在了build/opencl-build下面，记住此时的halide-install-path
make install -j32