Minor update

awslabs · Jan 13, 2024 · a904722 · a904722
1 parent f8ab9df
commit a904722
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/python/graphstorm/utils.py b/python/graphstorm/utils.py
@@ -29,7 +29,6 @@
 
 TORCH_MAJOR_VER = int(th.__version__.split('.', maxsplit=1)[0])
 USE_WHOLEGRAPH = False
-USE_WHOLEGRAPH_SPARSEEMB = False
 
 def setup_device(local_rank):
     r"""Setup computation device.
@@ -97,12 +96,6 @@ def is_wholegraph():
     """
     return USE_WHOLEGRAPH
 
-
-def is_wholegraph_sparse_emb():
-    """Check whether global USE_WHOLEGRAPH_SPARSEEMB is true"""
-    return USE_WHOLEGRAPH_SPARSEEMB
-
-
 def estimate_mem_train(root, task):
     ''' Estimate the memory consumption per machine during training.