缺失算子RROIAlign，且从pytorch的aten算子迁移至mindspore的aot算子失败 - Githubissues

mindspore-ai / mindspore

MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.

https://gitee.com/mindspore/mindspore

Apache License 2.0

4.3k stars 711 forks source link

缺失算子RROIAlign，且从pytorch的aten算子迁移至mindspore的aot算子失败 #207

Open BingyuanZhang opened 2 years ago

BingyuanZhang commented 2 years ago

Background

我在复现一个模型DRRG的过程中，发现需要一个算子RROIAlign，这个算子并不在torch和mindspore中，代码作者采用torch的Aten方式开发了一个算子，地址

Introduction

之后我在mindspore的官方文档中看到有使用aot方式进行开发算子的方法。不过因为mindspore和torch的tensor不一样，所以需要在CUDA中把 MindSpore kernel 的 inputs/outputs 转换为 PyTorch Aten 的 Tensor，但是我发现DRRG中开发算子的CUDA文件的格式和上述mindspore文档中的CUDA文件格式不一样，且我对CUDA的语法不了解，所以无从下手，不能完成转换。
希望官方可以实现一个RROIAlign的算子

cxl-ll commented 2 years ago

你好，问题正在分析中，会尽快解答回复，谢谢支持

BingyuanZhang commented 2 years ago

你好，问题正在分析中，会尽快解答回复，谢谢支持

您好，请问多久能解决这个问题呢？