已关闭
[Course-Feedback|课程反馈]: as_strided 测试脚本没有真正调用自定义算子 #385
sinobr创建于  7月31日关闭于  9月7日
sinobr
sinobr
7月31日 创建

我看 skills/examples/as_strided/run_test.py 时发现,测试脚本现在好像没有真正测到自己编译的算子。

run_as_strided_npu 里:

  • stride 非负时直接调用 torch.as_strided(input_tensor, size, stride, storage_offset);
  • stride 为负时直接调用 as_strided_golden(input_x, size, stride, storage_offset)。

主函数虽然在 143~149 行加载了 libcust_opapi.so 和 libcust_opmaster_rt2.0.so,但是 run_as_strided_npu 里没有看到 torch.ops 或 torch.classes 的自定义算子调用。

我把这个文件的调用顺了一遍,正 stride 走的是 PyTorch 自带的 torch.as_strided,负 stride 走的是 CPU golden,编译出来的 so 只是被加载,没有参与计算。这样即使 custom op 的 kernel 写错,测试也可能显示 PASS。

是不是应该让测试真正调用注册的自定义算子,再用 as_strided_golden 做结果对比?

likedislike
ink_polymer
ink_polymer成员
7月31日 评论:

尊敬的开发者您好,您的建议已收到,待内部评估后反馈,感谢您的支持

likedislike
Saturday22266Saturday22266
8月11日 关联了pull request:fix: 修复自定义算子未调用问题
Saturday22266Saturday22266
8月12日 关联了pull request:fix: 修复自定义算子未调用问题
ink_polymer
ink_polymer成员
9月1日 评论:

尊敬的开发者您好,您提出的问题目前已经解决。该issue将在7天后关闭。若后续还有问题欢迎通过issue与我们交流,感谢您的支持

likedislike
ink_polymerink_polymer成员
9月7日 issue状态由 进行中 改变为 已完成
ink_polymerink_polymer成员
9月7日 关闭了 issue
CANN-robotCANN-robot成员
9月7日 添加了label:resolved