From 4b5cb22fb9d3aa034866acfdc54bd1843684c228 Mon Sep 17 00:00:00 2001
From: zhulei <563755780@qq.com>
Date: Mon, 12 Apr 2021 17:12:39 +0800
Subject: [PATCH] [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
---
 python/paddle/tensor/random.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/paddle/tensor/random.py b/python/paddle/tensor/random.py
index ba7ca417382..5aca87c1507 100644
--- 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)
-- 
GitLab