Skip to content

Add a checked registry for engine environment variables - #800

Open
ZacharyZcR wants to merge 5 commits into
JustVugg:devfrom
ZacharyZcR:feat/env-registry
Open

Add a checked registry for engine environment variables#800
ZacharyZcR wants to merge 5 commits into
JustVugg:devfrom
ZacharyZcR:feat/env-registry

Conversation

@ZacharyZcR

@ZacharyZcR ZacharyZcR commented Aug 3, 2026

Copy link
Copy Markdown
Contributor

The problem

Environment-variable mistakes are currently silent:

$ COLI_PREFIL_CHUNK=512 ./colibri ...    # typo
$ K3_BITS=8 ./colibri ...                # wrong engine

Both runs proceed with defaults and can produce plausible but incorrectly labelled benchmark results. The tree had no authoritative list of the variables consumed across its engines.

What this adds

c/coli_env.h provides one checked registry containing each variable's name, value shape, owning engine, and deprecation state. coli_env_check() runs once at engine startup:

[env] unknown variable COLI_PREFIL_CHUNK -- did you mean COLI_PREFILL_CHUNK?
[env] K3_BITS is not read by colibri (it belongs to kimi_k3) -- it will have no effect

The default remains warning-only to preserve scripts that export knobs for multiple engines. COLI_ENV_STRICT=1 makes these errors fatal, while COLI_ENV_DUMP=1 prints the variables read by the selected engine and their current values.

Only project-owned prefixes and legacy project variables are checked; unrelated environment variables are ignored.

Drift prevention

This PR deliberately does not rewrite every existing getenv() call site. That mechanical migration is larger and can silently change defaults, so it should be handled separately, engine by engine.

make -C c check-env and its CI job keep the registry aligned with the source. The check rejects:

  • a getenv() with no registry row
  • a registry row that no source reads
  • duplicate names
  • an unsorted table, which would break coli_env_find() binary search

The registry now covers the current GLM, DeepSeek V4, Inkling, Kimi K3, OLMoE, and Qwen3.6 code paths. The exact count is intentionally left out of the title because it changes as engines evolve; the checker reports the live value.

Compatibility and deprecation

Unknown variables warn rather than exit unless strict mode is enabled. This preserves the existing runtime contract requested in review.

Only TEMP is marked deprecated. SNAP, SNAP_MIRROR, and PROMPT remain accepted without warnings because they are active spellings in documentation and CI.

Compiled objects are not part of this branch.

Verification after rebasing onto current dev

  • make -C c check-env: registry matches sources, 338 variables
  • make -C c test-c: passed
  • Python suite: 630 passed
  • CI env-registry check: passed
  • Linux CPU, container, Vulkan, all-engine Linux/macOS/Windows, CUDA/HIP syntax, Web UI, Python, ARM, and oracle jobs: passed

Windows UCRT64 and sanitizer jobs were still running when this description was refreshed; no current check had a terminal failure.

GPU runtime execution was not performed locally; GPU backend coverage is compile/syntax validation only.

Suggested follow-ups (not in this PR)

  1. Replace direct getenv() sites with registry lookups one engine at a time.
  2. Migrate collision-prone legacy bare names under project prefixes with a deprecation cycle.
  3. Decide separately whether disabled experimental knobs should remain registered or be removed.

@ZacharyZcR
ZacharyZcR marked this pull request as ready for review August 3, 2026 21:11
@ZacharyZcR

Copy link
Copy Markdown
Contributor Author

Rebased on current dev, and the rebase made the argument for this PR better than the description did.

make check-env came back red immediately:

coli_env.h is out of sync with the sources:
  - read by the code but missing from coli_env.h: COLI_MTP_GUARD_PCT (colibri.c:6091)
  - read by the code but missing from coli_env.h: COLI_MTP_GUARD_WINDOW (colibri.c:6092)
  - read by the code but missing from coli_env.h: COLI_V4_EXPERT_PREFETCH (deepseek_v4.c:3132)
  - read by the code but missing from coli_env.h: CUDA_EXPERT_LOAD_BALANCE (colibri.c:8078)
  - read by the code but missing from coli_env.h: INK_METAL_SHARED (inkling.c:1282)
  - read by the code but missing from coli_env.h: V4_PREFIX_LOG (deepseek_v4.c:6933)

Six variables in two days, from three different engines, none of them mine. That is the drift rate this exists to catch, measured rather than argued — and it is the reason a registry has to be checked by a machine instead of maintained by intention.

Types are taken from the call sites rather than the names: the three read through atoi() are CE_INT; INK_METAL_SHARED and COLI_V4_EXPERT_PREFETCH are tested as switches, so CE_BOOL; V4_PREFIX_LOG is presence-checked exactly like the INK_PREFIX_LOG row already in the table, so CE_STR for consistency.

One judgement call worth flagging

Two of the six belong to deepseek_v4.c, which #165 introduced and which has no engine bit in this table. I added CE_DSV4 — and deliberately did not fold it into CE_ALL.

CE_ALL means the engines that share route_trace.h, rans.h and omp_tune.h. deepseek_v4.c includes none of the three, so putting it in CE_ALL would silently assert that every shared-header knob is read by V4 as well. The flag would stop meaning anything. Say the word if you would rather have it in there.

Stale numbers in the header comment

The comment said "212 environment variables from 187 scattered getenv() call sites" — both already wrong, in precisely the way the table would have been without this check. The count is now 220. The 187 is removed rather than corrected: nothing verifies it, so it can only rot again. What is left points at make check-env, which prints the live count and is the one number in that comment that cannot go stale.

The PR title still says 214; it is 220 now. Happy to retitle, or leave it since the count moves with dev anyway.

Verified: check-env: 220 variables, registry matches the sources, colibri builds, make test-c green.

@JustVugg

JustVugg commented Aug 7, 2026

Copy link
Copy Markdown
Owner

Same blocker as #849, from the same cause — the branch carries the identical 16 compiled objects:

c/COLI_V4_UNIT_*.o  (12)
c/build/ownership/COLI_V4_UNIT_{CONFIG,NATIVE_QUANT,RUNTIME,ST}.o  (4)

.gitignore lists every engine binary except the one added last, so make deepseek-v4 leaves 26 untracked files that any git add -A picks up. #868 adds the missing entries. After it lands, a rebase drops them by itself; before it, git rm --cached c/COLI_V4_UNIT_*.o c/build/ownership/*.o.

The registry itself I want to read properly rather than wave through — 214 environment variables is most of the engine's real interface, and a check that rejects an unknown name is the kind of thing that is excellent until the day someone's working COLI_ variable stops being accepted. I will come back to it on its merits once the objects are out.

@JustVugg JustVugg added the enhancement New feature or request label Aug 7, 2026
@JustVugg

JustVugg commented Aug 7, 2026

Copy link
Copy Markdown
Owner

Two separate things here, and I want to check the second rather than assume.

1. The objects clear themselves now. #868 is on dev and added c/COLI_V4_UNIT_*.o (plus the deepseek_v4 binaries) to .gitignore — that is where all 16 came from. A rebase drops them. As I said on #849: the same thing happened to me on this repo shortly after I wrote that comment, so it is a repo gap, not a discipline one.

2. The body still opens with Draft. — which, going by #801 and #804, is how you mark work in progress rather than the GitHub flag. So I am reading this as not ready and leaving it alone, but I would rather ask than guess: is it still draft, or is that line stale?

Asking because I have got this wrong before in the other direction. I merged #804 and #806 without noticing the same marker in the body and had to say so publicly on #801. I would rather be the person who asks a redundant question than the one who does that twice.

If it is ready, rebase and say so and I will review it properly. On the substance I want to read it carefully rather than wave it through: 214 environment variables is most of the engine's real interface, and a registry that rejects unknown names is excellent right up until the day it rejects someone's working COLI_-prefixed variable. What I will be looking for is what happens on an unrecognised name — refuse, or warn and continue — and whether that is the same answer for a name the registry has never heard of as for one it knows is misspelled.

JustVugg pushed a commit that referenced this pull request Aug 7, 2026
#868 closed most of this: c/COLI_V4_UNIT_*.o covers the twelve objects the
amalgamated deepseek_v4 Makefile leaves next to the sources. Four more go
somewhere else.

    c/Makefile:895:  V4_OWN_DIR = build/ownership

so `make check` also writes build/ownership/COLI_V4_UNIT_{RUNTIME,CONFIG,
ST,NATIVE_QUANT}.o, which no rule matched. On current dev a clean checkout
plus `make check` still leaves `?? c/build/` in git status -- one
`git add -A` away from the same accident #868 was written to stop, and the
one I made on #849 and #800.

Ignoring c/build/ wholesale rather than the four names: nothing under it is
tracked, and it is a build output directory, so a rule per object would need
editing every time the ownership suite grows.

Worth stating because it caught me on #849: an ignore rule does not untrack
what is already committed. #868 made the twelve invisible in git status while
leaving them in the tree; git rm --cached is what removed them. Same shape as
#835. This prevents the next one, it does not clean up an existing one.
@ZacharyZcR

Copy link
Copy Markdown
Contributor Author

Ready for review. Rebased onto current dev; the compiled objects dropped out with the updated ignore rules. The rebase added six environment variables, now registered in the same table. Verified make -C c check-env (242 variables, registry matches the sources) and a clean default make -C c -j2 build.

Two failure modes, both silent today:

    COLI_PREFIL_CHUNK=512 ./colibri ...    # typo
    K3_BITS=8 ./colibri ...                # wrong engine

Neither prints anything. The run proceeds on the default, produces entirely
plausible output, and the number that gets written into a benchmark table is
for a configuration nobody set. Anyone who has tuned this engine has lost
an afternoon to one of these.

The cause is that nothing in the tree knew which variables exist: 214 of
them, read from 187 scattered getenv() sites, with no list anywhere. So:

  c/coli_env.h -- one table: name, value shape, which engines read it, and
  whether it is deprecated. coli_env_check() runs once at startup in all
  four engines and reports what it cannot use, with a "did you mean X?"
  from an edit-distance match:

    [env] unknown variable COLI_PREFIL_CHUNK -- did you mean COLI_PREFILL_CHUNK?
    [env] K3_BITS is not read by colibri (it belongs to kimi_k3) -- it will have no effect

  It warns, it does not exit: an unrecognised variable has never stopped a
  run and making it fatal would break scripts that export a knob for
  whichever engine they may launch. COLI_ENV_STRICT=1 makes it fatal for
  those who want the guarantee. COLI_ENV_DUMP=1 prints every variable the
  engine reads with its current value, which answers "is my export actually
  arriving" without a debugger.

  Variables we do not own are never touched -- only COLI_*, K3_* and INK_*
  are checked, so an unrelated EDITOR in the environment is not our
  business.

This is the shape vLLM arrived at (vllm/envs.py): one declaration point, one
prefix convention, validation at the boundary. Worth recording that vLLM
carries 284 of these to our 214 -- the count was never the problem in either
project, the missing registry was.

WHAT THIS DELIBERATELY DOES NOT DO

It does not rewrite the 187 getenv() call sites. That is mechanical but
large, and one mistyped default in the middle of it silently changes engine
behaviour; it belongs in its own commits, engine by engine, with this table
as the precondition for doing it safely.

So the table could drift from the code the moment it lands -- which is what
`make -C c check-env` (and a CI job) prevents. It fails on a getenv() with
no row, a row nothing reads, a duplicate, or a table that is not sorted
(coli_env_find binary-searches it, so an unsorted table silently stops
finding valid names). All three failure modes were verified by breaking the
table on purpose and confirming the check catches each one. It caught its
own first bug during development: COLI_ENV_DUMP was added to the code and
not to the table.

Only TEMP is marked deprecated -- it is the one the docs already call a
deprecated alias, and the one that actually bites, since $TEMP is the
temp-DIRECTORY path on Windows and under the ROCm runtime (JustVugg#509). SNAP,
SNAP_MIRROR and PROMPT are fallbacks but remain the mainstream spelling
(README, CONTRIBUTING's oracle command and CI all use SNAP=), so flagging
them would have printed a warning on every existing run and every CI job for
no benefit. Verified: today's usage is completely silent.

All four engines build warning-free and parse under every backend
configuration (default, CUDA, Vulkan, Metal, ANS); make test-c passes; the
Python suite is 288 tests OK (18 skipped).
Rebasing on dev turned check-env red, which is the point of it:

  - read by the code but missing from coli_env.h: COLI_MTP_GUARD_PCT
  - read by the code but missing from coli_env.h: COLI_MTP_GUARD_WINDOW
  - read by the code but missing from coli_env.h: COLI_V4_EXPERT_PREFETCH
  - read by the code but missing from coli_env.h: CUDA_EXPERT_LOAD_BALANCE
  - read by the code but missing from coli_env.h: INK_METAL_SHARED
  - read by the code but missing from coli_env.h: V4_PREFIX_LOG

Types come from the call sites, not from the names: the three read
through atoi() are CE_INT, INK_METAL_SHARED and COLI_V4_EXPERT_PREFETCH
are tested as switches so CE_BOOL, and V4_PREFIX_LOG is presence-checked
like the INK_PREFIX_LOG row already in the table, so CE_STR.

Two of them belong to deepseek_v4.c, which JustVugg#165 added and which has no
engine bit here, so add CE_DSV4. It is deliberately NOT folded into
CE_ALL: that flag means the four engines sharing route_trace.h, rans.h
and omp_tune.h, and deepseek_v4.c includes none of the three -- putting
it in CE_ALL would claim every shared-header knob is read by V4 too.

Also refresh the header comment, which had gone stale in the same way the
table would have: 212 -> 220. The "187 scattered getenv() call sites"
figure is removed rather than corrected -- nothing verifies it, so it can
only rot again. `make check-env` prints the live count and is the one
number in that comment that cannot go stale.

check-env: 220 variables, registry matches the sources.
Rebasing again turned check-env red again, which is the argument for the
PR restating itself: COLI_V4_{AUTOPIN,DIRECT,MARKOV_BLOCK,MARKOV_KEEP,
MARKOV_SPEC,PREWARM,SAVE_USAGE}, K3_CUDA, V4_{DRAFT,MTP,MTP_DRAFT,MTP_GB,
MTP_MIN,MTP_PARTIAL_KEEP,NGRAM,NGRAM_PARTIAL_KEEP} -- most of them from
JustVugg#839's MTP/DSpark work, landed since the last rebase.

Types from the call sites: the ones tested as atoi()!=0 or atoi()==0
switches are CE_BOOL, V4_MTP_GB parses with atof() and defaults to 0.45
so CE_FLOAT, the rest read a number with atoi() so CE_INT. K3_CUDA is
CE_KIMI; every V4_/COLI_V4_ row is CE_DSV4.

Header count 220 -> 236.
@ZacharyZcR ZacharyZcR changed the title Add a registry for the 214 environment variables, and check the environment against it Add a checked registry for engine environment variables Aug 28, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants