DistABLPLoader: vectorized label remap + optional AnchorLabels edge-list output by kmontemayor2-sc · Pull Request #681 · Snapchat/GiGL

Commits on Jun 25, 2026

refactor(ablp): extract per-anchor label loop into _loop_set_labels oracle

kmontemayor
and
claude
committed
feat(ablp): add vectorized_set_labels kernel, equivalence-tested vs loop oracle

kmontemayor
and
claude
committed
feat(ablp): add AnchorLabels edge-list container + edge_list_set_labels kernel

kmontemayor
and
claude
committed
test(ablp): add CUDA device-placement regression for label-remap kernels

kmontemayor
and
claude
committed
feat(ablp): vectorized label remap always + use_list_output ctor flag

kmontemayor
and
claude
committed
docs(ablp): document vectorized remap, use_list_output, and AnchorLabels

kmontemayor
and
claude
committed
docs(examples): consume AnchorLabels edge-list in homogeneous training

kmontemayor
and
claude
committed
docs(examples): consume AnchorLabels edge-list in heterogeneous training

kmontemayor
and
claude
committed
docs(examples): consume AnchorLabels edge-list in graph-store homogeneous training

kmontemayor
and
claude
committed
docs(examples): consume AnchorLabels edge-list in graph-store heterogeneous training

kmontemayor
and
claude
committed
feat(distributed): export AnchorLabels from gigl.distributed

kmontemayor
and
claude
committed
style(ablp): apply ruff formatting to ABLP label-output examples and test

kmontemayor
and
claude
committed
refactor(ablp): drop order-reproduction argsort; add readability refactors

kmontemayor
and
claude
committed
docs(examples): clarify AnchorLabels read is loss-equivalent, not order-identical

kmontemayor
and
claude
committed

Commits on Jun 26, 2026

refactor(ablp): single AnchorLabels label-remap kernel + dict view

kmontemayor
and
claude
committed
docs(ablp): why-first docstrings, worked examples, tensor-dim annotations

kmontemayor
and
claude
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

DistABLPLoader: vectorized label remap + optional AnchorLabels edge-list output - #681

DistABLPLoader: vectorized label remap + optional AnchorLabels edge-list output#681
kmontemayor2-sc wants to merge 20 commits into
mainfrom
kmonte/ablp-vectorized-labels-and-list-output

Commits on Jun 25, 2026

refactor(ablp): extract per-anchor label loop into _loop_set_labels oracle

feat(ablp): add vectorized_set_labels kernel, equivalence-tested vs loop oracle

feat(ablp): add AnchorLabels edge-list container + edge_list_set_labels kernel

test(ablp): add CUDA device-placement regression for label-remap kernels

feat(ablp): vectorized label remap always + use_list_output ctor flag

docs(ablp): document vectorized remap, use_list_output, and AnchorLabels

docs(examples): consume AnchorLabels edge-list in homogeneous training

docs(examples): consume AnchorLabels edge-list in heterogeneous training

docs(examples): consume AnchorLabels edge-list in graph-store homogeneous training

docs(examples): consume AnchorLabels edge-list in graph-store heterogeneous training

feat(distributed): export AnchorLabels from gigl.distributed

style(ablp): apply ruff formatting to ABLP label-output examples and test

refactor(ablp): drop order-reproduction argsort; add readability refactors

docs(examples): clarify AnchorLabels read is loss-equivalent, not order-identical

Commits on Jun 26, 2026

refactor(ablp): single AnchorLabels label-remap kernel + dict view

docs(ablp): why-first docstrings, worked examples, tensor-dim annotations

Commits on Jun 29, 2026

docs(ablp): worked example for use_list_output=True; harden node-map uniqueness guard

docs(ablp): scrub history-relative comments in label-remap code

Commits on Jul 28, 2026

refactor: represent ABLP labels as edge index tensors

fix(ablp): key label node map off label edge types, not supervision edge types