dataframe: passing device to PHETensor convertion

Signed-off-by: mgqa34 <mgq3374541@163.com> Signed-off-by: weiwee <wbwmat@gmail.com>
FederatedAI · Aug 9, 2023 · 0e6c30f · 0e6c30f
1 parent 070bf26
commit 0e6c30f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/fate/arch/dataframe/manager/block_manager.py b/python/fate/arch/dataframe/manager/block_manager.py
@@ -377,6 +377,7 @@ def convert_to_phe_tensor(self, block, shape):
                          coder=self._coder,
                          shape=shape,
                          data=block,
+                         device=self._device,
                          dtype=self._dtype)
 
     @property