def wait(self):
                """Returns the stored value when invoked."""
                gpu_cache = self.gpu_cache
                values, missing_index, missing_keys = (
                    self.future.wait() if async_op else self.future
                )
                # Ensure there is no leak.
                self.gpu_cache = self.future = None

                gpu_cache.total_queries += values.shape[0]
                gpu_cache.total_miss += missing_keys.shape[0]
                return values, missing_index, missing_keys
Returns the stored value when invoked.
python/dgl/graphbolt/impl/gpu_feature_cache.py