From 2ad151bee3427ab2599655aa078a198bb0892ce3 Mon Sep 17 00:00:00 2001 From: danleifeng <52735331+danleifeng@users.noreply.github.com> Date: Tue, 28 Mar 2023 16:16:09 +0800 Subject: [PATCH] fix ps gpu dense; (#241) --- python/paddle/distributed/ps/the_one_ps.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/paddle/distributed/ps/the_one_ps.py b/python/paddle/distributed/ps/the_one_ps.py index 97a85d19873b83..48d60c3364b53d 100755 --- 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)))