[Triton课程笔记] 2.1.4 Backend-build
一、build
- 写CMakeLists.txt文件;
- 写cmake/TritonYourBackendConfig.cmake.in 文件;
- 将pytorch_backend/src/libtriton_pytorch.ldscript 拷贝到your_backend/src/libtriton_yourbackend.ldscript;
- build:
- 单独build backend:
- 用CMake和make构建;
- 将构建好的backend复制到/opt/tritonserver/backends/yourbackend目录。
- 整个triton一起编译:
- 修改build.py,按需求增加opts;
- 运行build.py来编译Triton,加上--backend your_backend:<container_tag> 参数。
二、总结
本文版权归作者(https://www.cnblogs.com/harrymore/)和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文链接,如有问题, 可邮件(harrymore@126.com)咨询.