[Question]: ROCM平台不支持paddle.multinomial

PaddlePaddle / PaddleNLP

👑 Easy-to-use and powerful NLP and LLM library with 🤗 Awesome model zoo, supporting wide-range of NLP tasks from research to industrial applications, including 🗂Text Classification, 🔍 Neural Search, ❓ Question Answering, ℹ️ Information Extraction, 📄 Document Intelligence, 💌 Sentiment Analysis etc.

https://paddlenlp.readthedocs.io

Apache License 2.0

11.96k stars 2.91k forks source link

[Question]: ROCM平台不支持paddle.multinomial #5202

Open liucong521 opened 1 year ago

liucong521 commented 1 year ago

请提出你的问题

AssertionError: multinomial op is not supported on ROCM yet. 请问针对于在rocm平台上不支持paddle.multinomial()有什么替代方案么？

FrostML commented 1 year ago

当前 PaddlePaddle 版本似乎没有特别合适替代方案，或者尝试基于比较初级的数学计算直接实现。 PaddlePaddle issue 可以先在这里再提一个 issue，说明下预期在 ROCm 上支持下 multinomial