[BugFix] Fix error attributes setting in FusedLinear unitest (#50359)

* first commit. * change host logic * fix code bugs * fix code error --------- Co-authored-by: zhangbopd <1299246947@qq.com>
PaddlePaddle · Mar 6, 2023 · 11a6149 · 11a6149
1 parent 12eff93
commit 11a6149
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_fused_gemm_epilogue_grad_op.py b/python/paddle/fluid/tests/unittests/test_fused_gemm_epilogue_grad_op.py
@@ -46,7 +46,7 @@ def setUp(self):
             'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
         }
 
-        self.attrs = {"activation": 'none'}
+        self.attrs = {"activation_grad": 'none'}
 
         DX, DY, DBias = get_outputs(
             self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -105,7 +105,7 @@ def setUp(self):
             'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
         }
 
-        self.attrs = {"activation": 'none'}
+        self.attrs = {"activation_grad": 'none'}
 
         _, DY, DBias = get_outputs(
             self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -164,7 +164,7 @@ def setUp(self):
             'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
         }
 
-        self.attrs = {"activation": 'none'}
+        self.attrs = {"activation_grad": 'none'}
 
         _, DY, _ = get_outputs(
             self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -219,7 +219,7 @@ def setUp(self):
             'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
         }
 
-        self.attrs = {"activation": 'none'}
+        self.attrs = {"activation_grad": 'none'}
 
         DX, DY, _ = get_outputs(
             self.inputs['DOut'], self.inputs['X'], self.inputs['Y']

diff --git a/python/paddle/fluid/tests/unittests/xpu/test_fused_gemm_epilogue_grad_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_fused_gemm_epilogue_grad_op_xpu.py
@@ -60,7 +60,7 @@ def init_data(self):
                 'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
             }
 
-            self.attrs = {"activation": 'none'}
+            self.attrs = {"activation_grad": 'none'}
 
             DX, DY, DBias = get_outputs(
                 self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -81,7 +81,7 @@ def init_data(self):
                 'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
             }
 
-            self.attrs = {"activation": 'none'}
+            self.attrs = {"activation_grad": 'none'}
 
             _, DY, DBias = get_outputs(
                 self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -96,7 +96,7 @@ def init_data(self):
                 'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
             }
 
-            self.attrs = {"activation": 'none'}
+            self.attrs = {"activation_grad": 'none'}
 
             _, DY, _ = get_outputs(
                 self.inputs['DOut'], self.inputs['X'], self.inputs['Y']
@@ -111,7 +111,7 @@ def init_data(self):
                 'Y': np.random.random((4, 128)).astype(self.dtype) - 0.5,
             }
 
-            self.attrs = {"activation": 'none'}
+            self.attrs = {"activation_grad": 'none'}
 
             DX, DY, _ = get_outputs(
                 self.inputs['DOut'], self.inputs['X'], self.inputs['Y']