fix ps gpu dense; (PaddlePaddle#241)

danleifeng · Sep 12, 2023 · 2ad151b · 2ad151b
1 parent 1c182ca
commit 2ad151b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/distributed/ps/the_one_ps.py b/python/paddle/distributed/ps/the_one_ps.py
@@ -970,7 +970,8 @@ def _get_tables(self):
                 else:
                     tables.append(globals()['SparseTable'](self.context, ctx))
             else:
-                tables.append(globals()['DenseTable'](self.context, ctx))
+                if not self.use_ps_gpu:
+                    tables.append(globals()['DenseTable'](self.context, ctx))
         self.tensor_tables = self._get_tensor_tables()
         tables.extend(self.tensor_tables)
         tables.append(globals()['BarrierTable'](self.context, len(tables)))