[Rocm] fix python test of multinomial (#32158)

* [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial

[Rocm] fix python test of multinomial (#32158)
* [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial * [Rocm] fix python test of multinomial
4b5cb22f · zhulei · GitHub · 8dacfb5e · 4b5cb22f
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

python/paddle/tensor/random.py python/paddle/tensor/random.py +3 -0

未找到文件。
--- a/python/paddle/tensor/random.py
+++ b/python/paddle/tensor/random.py
@@ -135,6 +135,9 @@ def multinomial(x, num_samples=1, replacement=False, name=None):

    """

+    assert core.is_compiled_with_rocm() == False, (
+        "multinomial op is not supported on ROCM yet.")
+
    if in_dygraph_mode():
        return core.ops.multinomial(x, 'num_samples', num_samples,
                                    'replacement', replacement)