path # lines of code benchmarks/api/bench_dist_neighbor_loader.py 137 benchmarks/api/bench_feature.py 106 benchmarks/api/bench_sampler.py 97 benchmarks/api/run_dist_bench.py 68 setup.py 64 graphlearn_torch/include/shm_queue.h 126 graphlearn_torch/include/tensor_map.h 14 graphlearn_torch/include/func_factory.h 51 graphlearn_torch/include/common.h 43 graphlearn_torch/include/sample_queue.h 28 graphlearn_torch/include/common.cuh 68 graphlearn_torch/include/types.h 47 graphlearn_torch/include/sampler.h 17 graphlearn_torch/include/graph.h 101 graphlearn_torch/include/subgraph_op_base.h 16 graphlearn_torch/include/hash_table.cuh 123 graphlearn_torch/include/stitch_sample_results.h 18 graphlearn_torch/include/inducer_base.h 28 graphlearn_torch/include/negative_sampler.h 15 graphlearn_torch/include/unified_tensor.cuh 67 graphlearn_torch/python/loader/neighbor_loader.py 60 graphlearn_torch/python/loader/transform.py 97 graphlearn_torch/python/loader/__init__.py 5 graphlearn_torch/python/loader/link_loader.py 114 graphlearn_torch/python/loader/node_loader.py 64 graphlearn_torch/python/loader/link_neighbor_loader.py 52 graphlearn_torch/python/loader/subgraph_loader.py 52 graphlearn_torch/python/distributed/dist_table_dataset.py 228 graphlearn_torch/python/distributed/dist_options.py 141 graphlearn_torch/python/distributed/dist_random_partitioner.py 396 graphlearn_torch/python/distributed/dist_neighbor_loader.py 51 graphlearn_torch/python/distributed/__init__.py 34 graphlearn_torch/python/distributed/dist_subgraph_loader.py 36 graphlearn_torch/python/distributed/dist_neighbor_sampler.py 636 graphlearn_torch/python/distributed/dist_server.py 183 graphlearn_torch/python/distributed/rpc.py 349 graphlearn_torch/python/distributed/dist_graph.py 85 graphlearn_torch/python/distributed/event_loop.py 59 graphlearn_torch/python/distributed/dist_feature.py 352 graphlearn_torch/python/distributed/dist_sampling_producer.py 285 graphlearn_torch/python/distributed/dist_loader.py 328 graphlearn_torch/python/distributed/dist_dataset.py 214 graphlearn_torch/python/distributed/dist_client.py 44 graphlearn_torch/python/distributed/dist_link_neighbor_loader.py 58 graphlearn_torch/python/distributed/dist_context.py 141 graphlearn_torch/python/data/graph.py 190 graphlearn_torch/python/data/vineyard_utils.py 94 graphlearn_torch/python/data/table_dataset.py 116 graphlearn_torch/python/data/__init__.py 7 graphlearn_torch/python/data/unified_tensor.py 56 graphlearn_torch/python/data/reorder.py 19 graphlearn_torch/python/data/feature.py 171 graphlearn_torch/python/data/dataset.py 344 graphlearn_torch/python/utils/exit_status.py 7 graphlearn_torch/python/utils/device.py 26 graphlearn_torch/python/utils/singleton.py 8 graphlearn_torch/python/utils/__init__.py 8 graphlearn_torch/python/utils/tensor.py 62 graphlearn_torch/python/utils/mixin.py 15 graphlearn_torch/python/utils/topo.py 46 graphlearn_torch/python/utils/units.py 18 graphlearn_torch/python/utils/common.py 157 graphlearn_torch/python/utils/build_glt.py 116 graphlearn_torch/python/__init__.py 10 graphlearn_torch/python/typing.py 46 graphlearn_torch/python/py_export_glt.cc 183 graphlearn_torch/python/py_export_v6d.cc 21 graphlearn_torch/python/sampler/base.py 240 graphlearn_torch/python/sampler/__init__.py 3 graphlearn_torch/python/sampler/neighbor_sampler.py 561 graphlearn_torch/python/sampler/negative_sampler.py 18 graphlearn_torch/python/partition/base.py 636 graphlearn_torch/python/partition/partition_book.py 39 graphlearn_torch/python/partition/__init__.py 4 graphlearn_torch/python/partition/frequency_partitioner.py 147 graphlearn_torch/python/partition/random_partitioner.py 46 graphlearn_torch/python/channel/base.py 14 graphlearn_torch/python/channel/mp_channel.py 10 graphlearn_torch/python/channel/remote_channel.py 82 graphlearn_torch/python/channel/__init__.py 4 graphlearn_torch/python/channel/shm_channel.py 24 graphlearn_torch/csrc/cuda/inducer.cuh 78 graphlearn_torch/csrc/cuda/subgraph_op.cuh 49 graphlearn_torch/csrc/cuda/hash_table.cu 96 graphlearn_torch/csrc/cuda/graph.cu 127 graphlearn_torch/csrc/cuda/random_sampler.cu 341 graphlearn_torch/csrc/cuda/unified_tensor.cu 366 graphlearn_torch/csrc/cuda/random_sampler.cuh 29 graphlearn_torch/csrc/cuda/random_negative_sampler.cu 158 graphlearn_torch/csrc/cuda/random_negative_sampler.cuh 24 graphlearn_torch/csrc/cuda/stitch_sample_results.cu 94 graphlearn_torch/csrc/cuda/inducer.cu 303 graphlearn_torch/csrc/cuda/subgraph_op.cu 239 graphlearn_torch/csrc/cuda/weighted_sampler.cuh 30 graphlearn_torch/csrc/cpu/inducer.cc 158 graphlearn_torch/csrc/cpu/subgraph_op.h 27 graphlearn_torch/csrc/cpu/random_sampler.cc 148 graphlearn_torch/csrc/cpu/graph.cc 27 graphlearn_torch/csrc/cpu/weighted_sampler.cc 159 graphlearn_torch/csrc/cpu/weighted_sampler.h 34 graphlearn_torch/csrc/cpu/stitch_sample_results.cc 64 graphlearn_torch/csrc/cpu/random_sampler.h 31 graphlearn_torch/csrc/cpu/random_negative_sampler.cc 61 graphlearn_torch/csrc/cpu/subgraph_op.cc 68 graphlearn_torch/csrc/cpu/random_negative_sampler.h 18 graphlearn_torch/csrc/cpu/inducer.h 46 graphlearn_torch/csrc/shm_queue.cc 215 graphlearn_torch/csrc/tensor_map.cc 113 graphlearn_torch/csrc/sample_queue.cc 16 graphlearn_torch/v6d/vineyard_utils.cc 248 graphlearn_torch/v6d/vineyard_utils.h 42