feat: Named graph specializations in specializations.json (Prefill/Decode/Vision/Encoder/Embedding)#904

Open

vbaddi wants to merge 9 commits intoquic:mainfrom

vbaddi:feat/enabling_nested_new_spec_format

Commits on Apr 1, 2026

feat: emit named specializations format in specializations.json

vbaddi
authored and
quic-rishinr
committed
nit: get_compilation_dims: resolve the symbols dict transparently before reading fields

vbaddi
authored and
quic-rishinr
committed
nit: enable the named spec. for diffusers too

vbaddi
authored and
quic-rishinr
committed
nit: extend to all the VLMs

vbaddi
authored and
quic-rishinr
committed
nit: pecialization_module_name was popped inside the specializations JSON write block, fixing this

vbaddi
authored and
quic-rishinr
committed
nit: added a feature_len check in _infer_specialization_name before the generic seq_len==1 → Decode rule

vbaddi
authored and
quic-rishinr
committed
feat: replace specialization name inference with _graph_name tagging

vbaddi
authored and
quic-rishinr
committed
nit: update the review comments and add additional test case

vbaddi
authored and
quic-rishinr
committed
nit: skip the tests from the samplers in CI

vbaddi
authored and
quic-rishinr
committed