diff --git a/tools/get_quick_disable_lt.py b/tools/get_quick_disable_lt.py index 18ebdb0031747fb957fb77e078f7c4cd81142328..4805c909c1ba41dc49115ccec91ac23649922f2c 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