diff --git a/python/graphstorm/utils.py b/python/graphstorm/utils.py index f6156f5a1a..059ba2aa99 100644 --- 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.