Skip to content

Add a registry for the 214 environment variables, and check the environment against it - #800

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

Add a registry for the 214 environment variables, and check the environment against it#800
ZacharyZcR wants to merge 3 commits into
JustVugg:devfrom
ZacharyZcR:feat/env-registry

Conversation

@ZacharyZcR

Copy link
Copy Markdown
Contributor

Draft.

The problem

Two things are 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 ends up in a benchmark table is for a configuration nobody set.

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.

What this adds

c/coli_env.h — one table: name, value shape, which engines read it, whether it is deprecated. coli_env_check() runs once at startup in all four engines:

[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 suggestion is an edit-distance match against the table.

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 might 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.

On the count, since it is the obvious first question

This is the shape vLLM arrived at (vllm/envs.py): one declaration point, one prefix convention, validation at the boundary.

Worth recording what that file actually contains, because it inverts the intuition: vLLM carries 284 environment variables to our 214. It stays manageable through structure, not restraint — 94% under a single VLLM_ prefix (the exceptions are all borrowed conventions: CUDA_VISIBLE_DEVICES, LD_LIBRARY_PATH, NO_COLOR, S3_*), one dict as the single definition point, types declared under if TYPE_CHECKING:, env_with_choices() validators, and __getattr__ raising AttributeError on an unknown name.

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 change is mechanical but large, and one mistyped default in the middle of it silently alters engine behaviour. It belongs in its own commits, engine by engine, with this table as the precondition for doing it safely.

Which means the table could drift from the code the moment it lands. That 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 name
  • a table that is not sortedcoli_env_find() binary-searches it, so an unsorted table silently stops finding valid names

All four failure modes were verified by breaking the table on purpose and confirming the check catches each. It also caught its own first bug during development: COLI_ENV_DUMP was added to the code and not to the table.

On deprecation, and on not being noisy

Only TEMP is marked deprecated. It is the one the docs already call a deprecated alias and the one that actually bites — $TEMP is the temp-directory path on Windows and under the ROCm runtime (#509).

SNAP, SNAP_MIRROR and PROMPT are fallbacks in the code, and my first pass flagged them too. That was wrong: SNAP= is the mainstream spelling in the README, in CONTRIBUTING's oracle command and in CI itself, so flagging it would have printed a warning on every existing run and every CI job for no benefit. Verified explicitly — today's usage is completely silent, including the SNAP=tiny_inkling ./inkling line in inkling-oracle.

Verification

  • All four engines build warning-free
  • All four parse under every backend configuration: default, COLI_CUDA, COLI_VULKAN, COLI_METAL, COLI_ANS
  • make test-c passes; Python suite 288 tests OK (18 skipped)
  • make check-env: 214 variables, registry matches the sources
  • Behaviour checked on the real binaries, not just a stub: typo, wrong-engine, deprecated alias, unrelated variable, clean environment, and COLI_ENV_STRICT=1 exit code

Not verified here: no GPU path was run (no CUDA/Metal/Vulkan device available) — the backend configurations are compile-checked, not executed.

Suggested follow-ups (not in this PR)

  1. Replace the 187 getenv() sites with table lookups, one engine per commit.
  2. The 69 bare-name variables (TEMP, CHAT, HOT, WIDE, DIRECT, PIPE, CAP, SEED, …) are the ones that can genuinely collide in a real shell — TEMP already did. Renaming them under COLI_* needs a deprecation cycle, and the registry now has the machinery for it.
  3. EXPERT_BUDGET / EXPERT_BUDGET_EXPERIMENTAL is the one knob I can point at as a removal candidate: the code already disables it (measured empty operating window, issue #303). Whether it stays is a call for whoever owns that experiment.

@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.

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.
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.
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