From 3822247f425f532c1417b7ed565390e4769a275d Mon Sep 17 00:00:00 2001
From: "Lei.C" <773703646@qq.com>
Date: Fri, 9 Apr 2021 10:06:06 +0800
Subject: [PATCH] [ROCM] update rocm skip ut list, test=develop (#32149)

---
 tools/get_quick_disable_lt.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tools/get_quick_disable_lt.py b/tools/get_quick_disable_lt.py
index 18ebdb0031..4805c909c1 100644
--- a/tools/get_quick_disable_lt.py
+++ b/tools/get_quick_disable_lt.py
@@ -25,6 +25,13 @@ def download_file():
         url = "https://sys-p0.bj.bcebos.com/prec/{}".format('disable_ut_win')
     else:
         url = "https://sys-p0.bj.bcebos.com/prec/{}".format('disable_ut')
+    try:
+        import paddle.fluid.core as core
+        if core.is_compiled_with_rocm():
+            url = "https://sys-p0.bj.bcebos.com/prec/{}".format(
+                'disable_ut_rocm_ci')
+    except:
+        pass
     f = requests.get(url)
     data = f.text
     status_code = f.status_code
-- 
GitLab