CUDA

CUDA実装のONNXRuntimeカスタムオペレータを実装してみた

前回の続き。 前回の記事 ↓ maminus.hatenadiary.org 今回のソースコード ↓ github.com ありがたいことにGitHubのIssueでCUDA版のカスタムオペレータ実装方法について問い合わせをいただいたので時間ができた時に実装してみた。 ※ONNXRuntimeのカスタムオペ…