增加错误捕获类型 (#66223)

PaddlePaddle · Jul 22, 2024 · 454e6d9 · 454e6d9
1 parent 327672f
commit 454e6d9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/deprecated/legacy_test/test_clip_grad_norm_.py b/test/deprecated/legacy_test/test_clip_grad_norm_.py
@@ -78,7 +78,10 @@ def TestRuntimeErrorStaticMode():
             clip_grad_norm_(input_pd, max_norm=2, norm_type=float("inf"))
             paddle.disable_static()
 
-        self.assertRaises(RuntimeError, TestRuntimeErrorStaticMode)
+        if paddle.framework.use_pir_api():
+            self.assertRaises(AttributeError, TestRuntimeErrorStaticMode)
+        else:
+            self.assertRaises(RuntimeError, TestRuntimeErrorStaticMode)
 
 
 def run_test_equal(