昇腾社区首页
中文
注册

功能介绍

当前,部分算子开源仓中,采用了msOpGen提供的工程模板。然而,基于此模板进行算子调用较为复杂,且难以实现算子的轻量化调测。为了解决此类问题,我们可以利用msKPP工具提供的mskpp.tiling_funcmskpp.get_kernel_from_binary接口,直接调用msOpGen工程中的tiling函数以及用户自定义的Kernel函数。

  • 使用本功能时,算子输入输出仅支持numpy.Tensor、torch.Tensor。
  • 若CANN中曾经部署过相同类型的算子(op_type),用户修改了tiling函数并重新编译,则需要在CANN环境中重新部署该算子。