昇腾社区首页
中文
注册
开发者
下载

样例总结

非In-place算子在入图过程中存在一些注意事项,在实际操作时请注意:

  • 定义PyTorch算子时,Tensor类型输入在前,基本类型输入在后。
  • 以算子工程方式开发PyTorch算子的Ascend C实现,Ascend C算子名保证为PyTorch算子的大驼峰格式,同时入参顺序、类型与PyTorch算子一致。
  • 使用OpPlugin工程化适配能力,简化PyTorch算子Eager模式NPU适配层开发。