AUTHORS
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
csrc/common.cuh
csrc/kernels.cuh
csrc/turbo_gnn.cpp
csrc/gatv2/gatv2_kernel.cu
csrc/gt/graph_transformer.cu
csrc/reduction/reduction_aggr.cu
csrc/reduction/reduction_aggr_base.cu
csrc/spmm/cusparse_spmm.cpp
csrc/spmm/cusparse_spmm.h
csrc/spmm/edge_norm_kernels.cu
scripts/autotune.py
scripts/benchmark.py
scripts/kernel_launch_comparison.py
scripts/kernel_tune.py
scripts/run_profile.py
scripts/train.py
scripts/validate.py
src/__init__.py
src/backends/__init__.py
src/backends/autotune.py
src/backends/base.py
src/backends/registry.py
src/backends/cuda_backend/__init__.py
src/backends/cuda_backend/conv.py
src/backends/cuda_backend/gatv2_aggr/utils.py
src/backends/cuda_backend/gt_aggr/utils.py
src/backends/cuda_backend/reduction_aggr/utils.py
src/backends/cuda_backend/spmm_aggr/utils.py
src/backends/cuda_test_backend/__init__.py
src/backends/cuda_test_backend/conv.py
src/backends/cuda_test_backend/dot_aggr/utils.py
src/backends/cuda_test_backend/mean_aggr/utils.py
src/backends/cugraph_backend/__init__.py
src/backends/cugraph_backend/conv.py
src/backends/cusparse_backend/__init__.py
src/backends/cusparse_backend/conv.py
src/backends/cusparse_backend/utils.py
src/backends/dfgnn_backend/__init__.py
src/backends/dfgnn_backend/bindings.py
src/backends/dfgnn_backend/conv.py
src/backends/dgl_backend/__init__.py
src/backends/dgl_backend/conv.py
src/backends/dgl_backend/utils.py
src/backends/fusegnn_backend/__init__.py
src/backends/fusegnn_backend/conv.py
src/backends/fusegnn_backend/utils.py
src/backends/fusegnn_backend/convs/__init__.py
src/backends/fusegnn_backend/convs/gat_conv.py
src/backends/fusegnn_backend/convs/gcn_conv.py
src/backends/fusegnn_backend/functional/__init__.py
src/backends/fusegnn_backend/functional/aggregate.py
src/backends/fusegnn_backend/functional/format.py
src/backends/fusegnn_backend/functional/gat.py
src/backends/fusegnn_backend/functional/gcn.py
src/backends/pyg_backend/__init__.py
src/backends/pyg_backend/conv.py
src/backends/pyg_backend/utils.py
src/backends/tcgnn_backend/__init__.py
src/backends/tcgnn_backend/conv.py
src/backends/torch_native_backend/__init__.py
src/backends/torch_native_backend/conv.py
src/backends/triton_backend/__init__.py
src/backends/triton_backend/conv.py
src/backends/triton_backend/kernels_impl.py
src/benchmarking/__init__.py
src/benchmarking/autotuner.py
src/benchmarking/memory.py
src/benchmarking/microbench.py
src/benchmarking/profiler.py
src/data/__init__.py
src/data/converters.py
src/data/datasets.py
src/data/graphland_datasets.py
src/data/loaders.py
src/models/__init__.py
src/models/base.py
src/models/config.py
src/models/registry.py
src/models/architecture/__init__.py
src/models/architecture/classifier.py
src/models/architecture/encoder.py
src/models/layers/__init__.py
src/models/layers/blocks.py
src/models/layers/conv_dispatcher.py
src/training/__init__.py
src/training/hooks.py
src/training/metrics.py
src/training/optimizer.py
src/training/scheduler.py
src/training/trainer.py
src/utils/__init__.py
src/utils/checkpointing.py
src/utils/logger.py
src/utils/scripts_utils.py
src/utils/triton_constants.py
turbo_gnn/__init__.py
turbo_gnn/_autotune.py
turbo_gnn/_functions.py
turbo_gnn/_kernels.py
turbo_gnn/_timer.py
turbo_gnn/graph.py
turbo_gnn/ops.py
turbo_gnn.egg-info/PKG-INFO
turbo_gnn.egg-info/SOURCES.txt
turbo_gnn.egg-info/dependency_links.txt
turbo_gnn.egg-info/entry_points.txt
turbo_gnn.egg-info/requires.txt
turbo_gnn.egg-info/top_level.txt