archi829/bottleneck-oracle-graphs

Dataset Card — bottleneck-oracle-graphs Overview A synthetic graph dataset of PyTorch profiler execution traces converted into heterogeneous graphs, designed for GNN-based bottleneck prediction in transformer model inference. Each graph represents one forward pass of a transformer variant, with nodes labelled by whether they lie on the critical execution path. Dataset Construction Traces were generated by profiling three transformer configurations (tiny, small, medium) using torch.profiler… See the full description on the dataset page: https://huggingface.co/datasets/archi829/bottleneck-oracle-graphs.