算能科技
直播中

卞轮辉

7年用户 1041经验值
私信 关注
[问答]

如何使用BMLang开发自定义的算子?

如何使用BMLang开发自定义的算子

回帖(1)

王玉兰

2023-9-19 15:14:37
BMLang是算能科技面向用户推出的针对BM168x TPU的一套高级编程接口库,使得用户可以快速的基于TPU硬件开发自定义算子,甚至整个神经网络。BMLang的基本元素是:张量数据(bmlang::Tensor)和计算操作(bmlang::Operator)。用户需要使用bmlang::Tensor和bmlang::Operation来编写C++代码,然后在程序最后使用bmlang::compile或bmlang::compile_with_check来生成TPU可以运行的二进制文件bmodel,与普通网络编译产生的BMODEL文件一样,依赖于BMRuntime接口载入与执行。
举报

更多回帖

发帖
×
20
完善资料,
赚取积分