remove unused device fn.

dmlc · Jul 18, 2024 · 7fcc4b4 · 7fcc4b4
1 parent 3255817
commit 7fcc4b4
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/python/dgl/graphbolt/feature_store.py b/python/dgl/graphbolt/feature_store.py
@@ -43,11 +43,6 @@ def read_async(self, ids: torch.Tensor = None):
 
     def read_async_num_stages(self):
         """The number of stages of the read_async operation."""
-        return 1
-
-    @property
-    def device(self):
-        """The device this feature is stored at."""
         raise NotImplementedError
 
     def size(self):

diff --git a/python/dgl/graphbolt/impl/torch_based_feature_store.py b/python/dgl/graphbolt/impl/torch_based_feature_store.py
@@ -209,10 +209,6 @@ def size(self):
         """
         return self._tensor.size()[1:]
 
-    @property
-    def device(self):
-        return self._tensor.device
-
     def update(self, value: torch.Tensor, ids: torch.Tensor = None):
         """Update the feature store.
 
@@ -411,10 +407,6 @@ def size(self):
         """
         return self._tensor.size()[1:]
 
-    @property
-    def device(self):
-        return self._tensor.device
-
     def update(self, value: torch.Tensor, ids: torch.Tensor = None):
         """Disk based feature does not support update for now."""
         raise NotImplementedError