add group norm backward test condition

DeepLink-org · Nov 29, 2024 · 070ea65 · 070ea65
1 parent fe30ff7
commit 070ea65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/diopi_test/python/configs/diopi_configs.py b/diopi_test/python/configs/diopi_configs.py
@@ -7218,12 +7218,14 @@
             args=[
                 {
                     "ins": ["input"],
+                    "requires_grad": [True],
                     "shape": ((2, 256, 7, 10), (2, 256, 12, 12),
                             (12, 15, 8, 9),(3, 6, 9, 0)),
                     "dtype": [np.float32, np.float64, np.float16],
                 },
                 {
                     "ins": ["weight", "bias"],
+                    "requires_grad": [True],
                     "shape": ((256,), (12,),
                                (15,), (3,)),
                     "dtype": [np.float32, np.float64, np.float16],