@simulatte/doppler 0.1.6 → 0.1.8
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +145 -0
- package/README.md +16 -23
- package/package.json +30 -32
- package/src/adapters/adapter-registry.js +12 -1
- package/src/adapters/lora-loader.js +23 -6
- package/src/bridge/extension-client.d.ts +5 -0
- package/src/bridge/extension-client.js +40 -0
- package/src/bridge/index.d.ts +2 -1
- package/src/bridge/index.js +6 -4
- package/src/browser/browser-converter.js +31 -1
- package/src/browser/file-picker.js +6 -0
- package/src/browser/safetensors-parser-browser.js +84 -1
- package/src/browser/shard-io-browser.js +2 -2
- package/src/browser/tensor-source-download.js +8 -2
- package/src/browser/tensor-source-http.d.ts +1 -0
- package/src/browser/tensor-source-http.js +5 -1
- package/src/client/doppler-api.browser.js +20 -4
- package/src/client/doppler-api.js +19 -3
- package/src/client/doppler-provider/generation.js +12 -0
- package/src/client/doppler-provider/model-manager.d.ts +10 -0
- package/src/client/doppler-provider/model-manager.js +91 -19
- package/src/client/doppler-provider/source-runtime.d.ts +2 -1
- package/src/client/doppler-provider/source-runtime.js +132 -13
- package/src/client/doppler-registry.json +5 -20
- package/src/config/backward-registry-loader.js +17 -2
- package/src/config/execution-v0-contract-check.js +113 -15
- package/src/config/kernel-path-contract-check.js +57 -29
- package/src/config/kernel-path-loader.d.ts +5 -0
- package/src/config/kernel-path-loader.js +18 -36
- package/src/config/kernels/kernel-ref-digests.js +1 -1
- package/src/config/kernels/registry.js +14 -1
- package/src/config/kernels/registry.json +81 -5
- package/src/config/loader.d.ts +1 -1
- package/src/config/loader.js +15 -2
- package/src/config/merge-contract-check.js +66 -4
- package/src/config/merge-helpers.js +128 -7
- package/src/config/merge.d.ts +1 -0
- package/src/config/merge.js +10 -0
- package/src/config/param-validator.js +47 -2
- package/src/config/presets/kernel-paths/{gemma2-q4k-dequant-f32a.json → gemma2-q4k-dequant-f32a-nosubgroups.json} +3 -3
- package/src/config/presets/kernel-paths/gemma3-f16-fused-f32a-online-streamingprefill.json +223 -0
- package/src/config/presets/kernel-paths/{gemma3-q4k-dequant-f32a.json → gemma3-q4k-dequant-f32a-nosubgroups.json} +3 -3
- package/src/config/presets/kernel-paths/gemma3-q4k-dequant-f32w-f32a-online.json +56 -0
- package/src/config/presets/kernel-paths/lfm2-q4k-dequant-f32a-nosubgroups.json +61 -0
- package/src/config/presets/kernel-paths/registry.json +43 -8
- package/src/config/presets/models/gemma2.json +3 -2
- package/src/config/presets/models/gemma3.json +2 -0
- package/src/config/presets/models/qwen3.json +4 -3
- package/src/config/presets/models/qwen3_5.json +16 -0
- package/src/config/presets/runtime/experiments/bench/gemma3-bench-q4k.json +1 -1
- package/src/config/presets/runtime/experiments/debug/gemma3-debug-q4k.json +1 -1
- package/src/config/presets/runtime/experiments/verify/gemma3-verify.json +1 -1
- package/src/config/presets/runtime/kernels/dequant-f16-q4k.json +6 -13
- package/src/config/presets/runtime/kernels/dequant-f32-q4k.json +6 -13
- package/src/config/presets/runtime/kernels/embeddinggemma-q4k-dequant-f32a.json +37 -0
- package/src/config/presets/runtime/kernels/fused-q4k.json +6 -13
- package/src/config/presets/runtime/kernels/gemma2-q4k-dequant-f16a.json +33 -0
- package/src/config/presets/runtime/kernels/gemma2-q4k-dequant-f32a-nosubgroups.json +33 -0
- package/src/config/presets/runtime/kernels/gemma2-q4k-fused-f32a.json +33 -0
- package/src/config/presets/runtime/kernels/safe-q4k.json +6 -13
- package/src/config/presets/runtime/model/qwen3-5-layer-probe.json +52 -0
- package/src/config/presets/runtime/model/qwen3-5-linear-attn-debug.json +90 -0
- package/src/config/presets/runtime/platform/metal-apple-q4k.json +1 -1
- package/src/config/runtime.js +6 -1
- package/src/config/schema/conversion.schema.d.ts +1 -0
- package/src/config/schema/debug.schema.d.ts +5 -0
- package/src/config/schema/doppler.schema.js +16 -21
- package/src/config/schema/inference-defaults.schema.js +3 -3
- package/src/config/schema/kernel-path.schema.d.ts +5 -1
- package/src/config/schema/kernel-thresholds.schema.js +12 -4
- package/src/config/schema/manifest.schema.d.ts +3 -2
- package/src/config/schema/manifest.schema.js +17 -4
- package/src/config/schema/storage.schema.js +1 -1
- package/src/config/training-defaults.js +30 -22
- package/src/converter/conversion-plan.js +104 -11
- package/src/converter/core.d.ts +7 -0
- package/src/converter/core.js +16 -9
- package/src/converter/execution-v0-manifest.js +4 -1
- package/src/converter/index.d.ts +1 -0
- package/src/converter/index.js +1 -0
- package/src/converter/manifest-inference.js +50 -29
- package/src/converter/parsers/diffusion.js +0 -3
- package/src/converter/parsers/transformer.js +4 -0
- package/src/converter/quantization-info.js +40 -16
- package/src/converter/quantizer.js +19 -12
- package/src/converter/rope-config.js +8 -6
- package/src/converter/shard-packer.d.ts +1 -1
- package/src/converter/shard-packer.js +4 -1
- package/src/converter/tokenizer-utils.d.ts +1 -0
- package/src/converter/tokenizer-utils.js +4 -1
- package/src/debug/config.js +123 -11
- package/src/debug/reference/hf_qwen35_linear_attn_debug.py +268 -0
- package/src/debug/signals.js +7 -1
- package/src/debug/tensor.d.ts +2 -0
- package/src/debug/tensor.js +13 -2
- package/src/distribution/p2p-control-plane.js +52 -12
- package/src/distribution/p2p-observability.js +43 -7
- package/src/distribution/p2p-webrtc-browser.js +20 -0
- package/src/distribution/shard-delivery.js +83 -27
- package/src/formats/gguf/types.js +33 -16
- package/src/formats/rdrr/groups.d.ts +12 -4
- package/src/formats/rdrr/groups.js +3 -6
- package/src/formats/rdrr/parsing.d.ts +4 -0
- package/src/formats/rdrr/parsing.js +53 -3
- package/src/formats/rdrr/types.d.ts +2 -1
- package/src/gpu/command-recorder.js +86 -61
- package/src/gpu/device.d.ts +1 -0
- package/src/gpu/device.js +73 -19
- package/src/gpu/kernel-tuner/benchmarks.js +326 -316
- package/src/gpu/kernel-tuner/cache.js +71 -4
- package/src/gpu/kernel-tuner/tuner.js +22 -4
- package/src/gpu/kernels/attention.js +15 -34
- package/src/gpu/kernels/backward/adam.js +62 -58
- package/src/gpu/kernels/backward/attention_backward.js +257 -169
- package/src/gpu/kernels/backward/conv2d_backward.js +14 -1
- package/src/gpu/kernels/cast.js +191 -149
- package/src/gpu/kernels/check-stop.js +33 -44
- package/src/gpu/kernels/conv2d.js +27 -17
- package/src/gpu/kernels/cross_entropy_loss.js +21 -15
- package/src/gpu/kernels/depthwise_conv2d.js +36 -26
- package/src/gpu/kernels/dequant.js +178 -126
- package/src/gpu/kernels/energy.d.ts +3 -21
- package/src/gpu/kernels/energy.js +111 -88
- package/src/gpu/kernels/feature-check.js +1 -1
- package/src/gpu/kernels/fused_ffn.js +84 -65
- package/src/gpu/kernels/fused_matmul_residual.js +56 -33
- package/src/gpu/kernels/fused_matmul_rmsnorm.js +62 -45
- package/src/gpu/kernels/gather.js +33 -15
- package/src/gpu/kernels/gelu.js +19 -11
- package/src/gpu/kernels/grouped_pointwise_conv2d.js +33 -23
- package/src/gpu/kernels/groupnorm.js +34 -23
- package/src/gpu/kernels/index.d.ts +8 -0
- package/src/gpu/kernels/index.js +6 -0
- package/src/gpu/kernels/kv-quantize.js +5 -2
- package/src/gpu/kernels/layernorm.js +35 -19
- package/src/gpu/kernels/logit-merge.js +5 -3
- package/src/gpu/kernels/matmul-selection.js +47 -4
- package/src/gpu/kernels/matmul.d.ts +2 -0
- package/src/gpu/kernels/matmul.js +59 -40
- package/src/gpu/kernels/modulate.js +23 -15
- package/src/gpu/kernels/moe.js +221 -175
- package/src/gpu/kernels/pixel_shuffle.js +22 -14
- package/src/gpu/kernels/relu.js +18 -10
- package/src/gpu/kernels/repeat_channels.js +25 -17
- package/src/gpu/kernels/residual.js +37 -27
- package/src/gpu/kernels/rmsnorm.js +66 -43
- package/src/gpu/kernels/rope.js +3 -0
- package/src/gpu/kernels/sample.js +27 -38
- package/src/gpu/kernels/sana_linear_attention.js +18 -10
- package/src/gpu/kernels/scale.js +18 -11
- package/src/gpu/kernels/shader-cache.js +4 -2
- package/src/gpu/kernels/silu.js +120 -72
- package/src/gpu/kernels/softmax.js +44 -25
- package/src/gpu/kernels/split_qg.d.ts +50 -0
- package/src/gpu/kernels/split_qg.js +46 -0
- package/src/gpu/kernels/split_qg.wgsl +58 -0
- package/src/gpu/kernels/split_qg_f16.wgsl +62 -0
- package/src/gpu/kernels/split_qkv.js +23 -13
- package/src/gpu/kernels/transpose.js +18 -10
- package/src/gpu/kernels/transpose.wgsl +5 -3
- package/src/gpu/kernels/upsample2d.js +21 -13
- package/src/gpu/kernels/utils.js +20 -13
- package/src/gpu/partitioned-buffer-pool.js +10 -2
- package/src/gpu/perf-guards.js +2 -9
- package/src/gpu/profiler.js +27 -22
- package/src/gpu/readback-utils.d.ts +16 -0
- package/src/gpu/readback-utils.js +41 -0
- package/src/gpu/submit-tracker.js +13 -0
- package/src/gpu/uniform-cache.d.ts +1 -0
- package/src/gpu/uniform-cache.js +30 -9
- package/src/gpu/weight-buffer.d.ts +1 -1
- package/src/gpu/weight-buffer.js +1 -1
- package/src/hotswap/intent-bundle.js +6 -0
- package/src/hotswap/manifest.d.ts +10 -1
- package/src/hotswap/manifest.js +12 -2
- package/src/hotswap/runtime.js +30 -8
- package/src/index-browser.d.ts +44 -0
- package/src/index-browser.js +14 -0
- package/src/inference/browser-harness-contract-helpers.d.ts +5 -0
- package/src/inference/browser-harness-contract-helpers.js +28 -0
- package/src/inference/browser-harness-diffusion-energy-suites.d.ts +2 -0
- package/src/inference/browser-harness-diffusion-energy-suites.js +269 -0
- package/src/inference/browser-harness-model-helpers.d.ts +16 -0
- package/src/inference/browser-harness-model-helpers.js +217 -0
- package/src/inference/browser-harness-report-helpers.d.ts +7 -0
- package/src/inference/browser-harness-report-helpers.js +42 -0
- package/src/inference/browser-harness-runtime-helpers.d.ts +61 -0
- package/src/inference/browser-harness-runtime-helpers.js +415 -0
- package/src/inference/browser-harness-suite-helpers.d.ts +28 -0
- package/src/inference/browser-harness-suite-helpers.js +268 -0
- package/src/inference/browser-harness-text-helpers.d.ts +27 -0
- package/src/inference/browser-harness-text-helpers.js +788 -0
- package/src/inference/browser-harness.d.ts +8 -0
- package/src/inference/browser-harness.js +149 -1996
- package/src/inference/kv-cache/base.js +140 -94
- package/src/inference/kv-cache/tiered.js +5 -3
- package/src/inference/moe-router.js +88 -56
- package/src/inference/multi-model-network.js +5 -3
- package/src/inference/network-evolution.d.ts +11 -2
- package/src/inference/network-evolution.js +20 -21
- package/src/inference/pipelines/context.d.ts +3 -0
- package/src/inference/pipelines/context.js +142 -2
- package/src/inference/pipelines/diffusion/helpers.js +10 -2
- package/src/inference/pipelines/diffusion/pipeline.js +2 -1
- package/src/inference/pipelines/diffusion/sd3-transformer.js +10 -10
- package/src/inference/pipelines/diffusion/text-encoder-gpu.js +8 -2
- package/src/inference/pipelines/diffusion/vae.js +3 -7
- package/src/inference/pipelines/energy/pipeline.js +27 -21
- package/src/inference/pipelines/energy/quintel.d.ts +5 -0
- package/src/inference/pipelines/energy/quintel.js +11 -0
- package/src/inference/pipelines/energy-head/row-head-pipeline.js +17 -13
- package/src/inference/pipelines/structured/json-head-pipeline.js +26 -11
- package/src/inference/pipelines/text/attention/output-projection.d.ts +12 -0
- package/src/inference/pipelines/text/attention/output-projection.js +8 -0
- package/src/inference/pipelines/text/attention/projections.d.ts +10 -1
- package/src/inference/pipelines/text/attention/projections.js +192 -112
- package/src/inference/pipelines/text/attention/record.js +77 -14
- package/src/inference/pipelines/text/attention/run.js +112 -14
- package/src/inference/pipelines/text/config.js +17 -4
- package/src/inference/pipelines/text/embed.js +2 -8
- package/src/inference/pipelines/text/execution-plan.js +46 -23
- package/src/inference/pipelines/text/execution-v0-contract-helpers.d.ts +59 -0
- package/src/inference/pipelines/text/execution-v0-contract-helpers.js +937 -0
- package/src/inference/pipelines/text/execution-v0-runtime-builders.d.ts +15 -0
- package/src/inference/pipelines/text/execution-v0-runtime-builders.js +279 -0
- package/src/inference/pipelines/text/execution-v0.js +62 -1013
- package/src/inference/pipelines/text/generator-runtime.js +5 -0
- package/src/inference/pipelines/text/generator-steps.d.ts +52 -0
- package/src/inference/pipelines/text/generator-steps.js +340 -221
- package/src/inference/pipelines/text/generator.js +56 -40
- package/src/inference/pipelines/text/init.d.ts +13 -0
- package/src/inference/pipelines/text/init.js +94 -25
- package/src/inference/pipelines/text/kernel-path-auto-select.js +2 -0
- package/src/inference/pipelines/text/kernel-trace.d.ts +2 -0
- package/src/inference/pipelines/text/kernel-trace.js +6 -0
- package/src/inference/pipelines/text/layer.js +4 -9
- package/src/inference/pipelines/text/linear-attention.d.ts +15 -0
- package/src/inference/pipelines/text/linear-attention.js +113 -9
- package/src/inference/pipelines/text/logits/gpu.js +12 -7
- package/src/inference/pipelines/text/logits/index.d.ts +6 -1
- package/src/inference/pipelines/text/logits/index.js +13 -12
- package/src/inference/pipelines/text/logits/utils.d.ts +7 -0
- package/src/inference/pipelines/text/logits/utils.js +9 -0
- package/src/inference/pipelines/text/lora-apply.js +50 -32
- package/src/inference/pipelines/text/model-load.js +282 -104
- package/src/inference/pipelines/text/moe-cache.js +5 -4
- package/src/inference/pipelines/text/moe-cpu-gptoss.js +74 -69
- package/src/inference/pipelines/text/moe-cpu.js +42 -38
- package/src/inference/pipelines/text/moe-gpu.js +110 -86
- package/src/inference/pipelines/text/ops.js +90 -90
- package/src/inference/pipelines/text/probes.js +9 -9
- package/src/inference/pipelines/text/sampling.js +52 -6
- package/src/inference/pipelines/text/weights.js +17 -7
- package/src/inference/pipelines/text.js +13 -1
- package/src/inference/speculative.d.ts +2 -2
- package/src/inference/speculative.js +4 -18
- package/src/inference/test-harness.d.ts +1 -1
- package/src/inference/test-harness.js +17 -7
- package/src/inference/tokenizer.d.ts +0 -5
- package/src/inference/tokenizer.js +4 -23
- package/src/inference/tokenizers/bpe.js +9 -0
- package/src/inference/tokenizers/bundled.js +20 -0
- package/src/inference/tokenizers/sentencepiece.js +12 -0
- package/src/loader/doppler-loader.js +38 -22
- package/src/loader/dtype-utils.js +3 -44
- package/src/loader/embedding-loader.js +7 -3
- package/src/loader/experts/expert-cache.js +13 -6
- package/src/loader/experts/expert-loader.js +10 -6
- package/src/loader/final-weights-loader.js +10 -4
- package/src/loader/layer-loader.js +2 -1
- package/src/loader/loader-state.js +2 -2
- package/src/loader/memory-monitor.js +8 -0
- package/src/loader/multi-model-loader.d.ts +14 -0
- package/src/loader/multi-model-loader.js +70 -24
- package/src/loader/shard-cache.js +84 -14
- package/src/loader/shard-resolver.js +25 -3
- package/src/loader/tensors/tensor-loader.js +214 -144
- package/src/loader/tensors/tensor-reader.js +76 -19
- package/src/loader/weight-downcast.js +1 -1
- package/src/memory/buffer-pool.d.ts +9 -1
- package/src/memory/buffer-pool.js +109 -44
- package/src/memory/unified-detect.js +1 -1
- package/src/rules/inference/dtype.rules.json +5 -0
- package/src/rules/inference/kernel-path.rules.json +24 -8
- package/src/rules/kernels/split-qg.rules.json +6 -0
- package/src/rules/rule-registry.js +27 -1
- package/src/storage/backends/opfs-store.js +68 -24
- package/src/storage/downloader.js +365 -83
- package/src/storage/index.d.ts +3 -0
- package/src/storage/index.js +3 -0
- package/src/storage/preflight.d.ts +2 -2
- package/src/storage/preflight.js +24 -2
- package/src/storage/quickstart-downloader.js +11 -5
- package/src/storage/registry.js +10 -4
- package/src/storage/reports.js +1 -1
- package/src/storage/shard-manager.d.ts +15 -1
- package/src/storage/shard-manager.js +55 -6
- package/src/storage/source-artifact-store.d.ts +52 -0
- package/src/storage/source-artifact-store.js +234 -0
- package/src/tooling/command-api-constants.d.ts +9 -0
- package/src/tooling/command-api-constants.js +9 -0
- package/src/tooling/command-api-family-normalizers.d.ts +9 -0
- package/src/tooling/command-api-family-normalizers.js +343 -0
- package/src/tooling/command-api-helpers.d.ts +25 -0
- package/src/tooling/command-api-helpers.js +262 -0
- package/src/tooling/command-api.js +16 -602
- package/src/tooling/command-envelope.js +4 -1
- package/src/tooling/command-runner-shared.js +52 -18
- package/src/tooling/conversion-config-materializer.js +3 -5
- package/src/tooling/lean-execution-contract.js +150 -3
- package/src/tooling/node-browser-command-runner.js +161 -271
- package/src/tooling/node-command-runner.js +29 -3
- package/src/tooling/node-converter.js +30 -1
- package/src/tooling/node-source-runtime.d.ts +1 -1
- package/src/tooling/node-source-runtime.js +120 -3
- package/src/tooling/node-webgpu.js +24 -21
- package/src/tooling/opfs-cache.js +21 -4
- package/src/tooling/runtime-input-composition.d.ts +38 -0
- package/src/tooling/runtime-input-composition.js +86 -0
- package/src/tooling/source-runtime-bundle.d.ts +40 -5
- package/src/tooling/source-runtime-bundle.js +261 -34
- package/src/tooling/source-runtime-materializer.d.ts +6 -0
- package/src/tooling/source-runtime-materializer.js +93 -0
- package/src/training/attention-backward.js +32 -17
- package/src/training/autograd.js +80 -52
- package/src/training/checkpoint-watch.d.ts +2 -1
- package/src/training/checkpoint-watch.js +39 -6
- package/src/training/checkpoint.js +40 -11
- package/src/training/clip.js +2 -1
- package/src/training/datasets/token-batch.js +20 -8
- package/src/training/distillation/checkpoint-watch.js +1 -0
- package/src/training/distillation/student-fixture.d.ts +22 -0
- package/src/training/distillation/student-fixture.js +846 -0
- package/src/training/distillation/suite-data.d.ts +45 -0
- package/src/training/distillation/suite-data.js +189 -0
- package/src/training/lora-pipeline.js +4 -7
- package/src/training/lora.js +26 -12
- package/src/training/loss.js +5 -6
- package/src/training/objectives/cross_entropy.js +2 -5
- package/src/training/objectives/distill_kd.js +4 -8
- package/src/training/objectives/distill_triplet.js +4 -8
- package/src/training/objectives/ul_stage2_base.js +4 -8
- package/src/training/operator-command.js +2 -0
- package/src/training/optimizer.js +19 -7
- package/src/training/runner.js +2 -1
- package/src/training/suite.js +18 -978
- package/src/training/tensor-factory.d.ts +9 -0
- package/src/training/tensor-factory.js +13 -0
- package/src/training/trainer.js +3 -5
- package/src/training/ul_dataset.js +3 -5
- package/src/training/workloads.js +70 -79
- package/src/types/model.d.ts +5 -0
- package/src/version.js +1 -1
- package/tools/convert-safetensors-node.js +22 -16
- package/tools/doppler-cli.js +50 -26
|
@@ -1,20 +1,13 @@
|
|
|
1
1
|
{
|
|
2
2
|
"id": "kernels/dequant-f16-q4k",
|
|
3
3
|
"name": "dequant-f16-q4k",
|
|
4
|
-
"description": "
|
|
4
|
+
"description": "Deprecated alias for kernels/gemma2-q4k-dequant-f16a.",
|
|
5
5
|
"intent": "investigate",
|
|
6
|
-
"stability": "
|
|
6
|
+
"stability": "deprecated",
|
|
7
7
|
"owner": "doppler-core",
|
|
8
8
|
"createdAtUtc": "2026-02-25T00:00:00Z",
|
|
9
|
-
"
|
|
10
|
-
"
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
"kernelPathPolicy": {
|
|
14
|
-
"mode": "capability-aware",
|
|
15
|
-
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
16
|
-
"onIncompatible": "remap"
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
}
|
|
9
|
+
"deprecatedAtUtc": "2026-03-08T00:00:00Z",
|
|
10
|
+
"replacementId": "kernels/gemma2-q4k-dequant-f16a",
|
|
11
|
+
"extends": "kernels/gemma2-q4k-dequant-f16a",
|
|
12
|
+
"runtime": {}
|
|
20
13
|
}
|
|
@@ -1,20 +1,13 @@
|
|
|
1
1
|
{
|
|
2
2
|
"id": "kernels/dequant-f32-q4k",
|
|
3
3
|
"name": "dequant-f32-q4k",
|
|
4
|
-
"description": "
|
|
4
|
+
"description": "Deprecated alias for kernels/gemma2-q4k-dequant-f32a-nosubgroups.",
|
|
5
5
|
"intent": "investigate",
|
|
6
|
-
"stability": "
|
|
6
|
+
"stability": "deprecated",
|
|
7
7
|
"owner": "doppler-core",
|
|
8
8
|
"createdAtUtc": "2026-02-25T00:00:00Z",
|
|
9
|
-
"
|
|
10
|
-
"
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
"kernelPathPolicy": {
|
|
14
|
-
"mode": "capability-aware",
|
|
15
|
-
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
16
|
-
"onIncompatible": "remap"
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
}
|
|
9
|
+
"deprecatedAtUtc": "2026-03-08T00:00:00Z",
|
|
10
|
+
"replacementId": "kernels/gemma2-q4k-dequant-f32a-nosubgroups",
|
|
11
|
+
"extends": "kernels/gemma2-q4k-dequant-f32a-nosubgroups",
|
|
12
|
+
"runtime": {}
|
|
20
13
|
}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "kernels/embeddinggemma-q4k-dequant-f32a",
|
|
3
|
+
"name": "embeddinggemma-q4k-dequant-f32a",
|
|
4
|
+
"description": "Explicit runtime override preset for the EmbeddingGemma Q4K dequant-to-F32 kernel path.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-09T00:00:00Z",
|
|
9
|
+
"extends": "default",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"compute": {
|
|
13
|
+
"activationDtype": "f32"
|
|
14
|
+
},
|
|
15
|
+
"kvcache": {
|
|
16
|
+
"kvDtype": "f32"
|
|
17
|
+
},
|
|
18
|
+
"session": {
|
|
19
|
+
"compute": {
|
|
20
|
+
"defaults": {
|
|
21
|
+
"activationDtype": "f32",
|
|
22
|
+
"outputDtype": "f32"
|
|
23
|
+
}
|
|
24
|
+
},
|
|
25
|
+
"kvcache": {
|
|
26
|
+
"kvDtype": "f32"
|
|
27
|
+
}
|
|
28
|
+
},
|
|
29
|
+
"kernelPath": "embeddinggemma-q4k-dequant-f32a",
|
|
30
|
+
"kernelPathPolicy": {
|
|
31
|
+
"mode": "capability-aware",
|
|
32
|
+
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
33
|
+
"onIncompatible": "remap"
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
@@ -1,20 +1,13 @@
|
|
|
1
1
|
{
|
|
2
2
|
"id": "kernels/fused-q4k",
|
|
3
3
|
"name": "fused-q4k",
|
|
4
|
-
"description": "
|
|
4
|
+
"description": "Deprecated alias for kernels/gemma2-q4k-fused-f32a.",
|
|
5
5
|
"intent": "investigate",
|
|
6
|
-
"stability": "
|
|
6
|
+
"stability": "deprecated",
|
|
7
7
|
"owner": "doppler-core",
|
|
8
8
|
"createdAtUtc": "2026-02-25T00:00:00Z",
|
|
9
|
-
"
|
|
10
|
-
"
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
"kernelPathPolicy": {
|
|
14
|
-
"mode": "capability-aware",
|
|
15
|
-
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
16
|
-
"onIncompatible": "remap"
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
}
|
|
9
|
+
"deprecatedAtUtc": "2026-03-08T00:00:00Z",
|
|
10
|
+
"replacementId": "kernels/gemma2-q4k-fused-f32a",
|
|
11
|
+
"extends": "kernels/gemma2-q4k-fused-f32a",
|
|
12
|
+
"runtime": {}
|
|
20
13
|
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "kernels/gemma2-q4k-dequant-f16a",
|
|
3
|
+
"name": "gemma2-q4k-dequant-f16a",
|
|
4
|
+
"description": "Explicit runtime override preset for the Gemma 2 Q4K dequant-to-F16 kernel path.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-08T00:00:00Z",
|
|
9
|
+
"extends": "default",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"compute": {
|
|
13
|
+
"activationDtype": "f16"
|
|
14
|
+
},
|
|
15
|
+
"kvcache": {
|
|
16
|
+
"kvDtype": "f16"
|
|
17
|
+
},
|
|
18
|
+
"session": {
|
|
19
|
+
"compute": {
|
|
20
|
+
"defaults": {
|
|
21
|
+
"outputDtype": "f16"
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
},
|
|
25
|
+
"kernelPath": "gemma2-q4k-dequant-f16a",
|
|
26
|
+
"kernelPathPolicy": {
|
|
27
|
+
"mode": "capability-aware",
|
|
28
|
+
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
29
|
+
"onIncompatible": "remap"
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "kernels/gemma2-q4k-dequant-f32a-nosubgroups",
|
|
3
|
+
"name": "gemma2-q4k-dequant-f32a-nosubgroups",
|
|
4
|
+
"description": "Explicit runtime override preset for the Gemma 2 Q4K dequant-to-F32 no-subgroups kernel path.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-08T00:00:00Z",
|
|
9
|
+
"extends": "default",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"compute": {
|
|
13
|
+
"activationDtype": "f32"
|
|
14
|
+
},
|
|
15
|
+
"kvcache": {
|
|
16
|
+
"kvDtype": "f32"
|
|
17
|
+
},
|
|
18
|
+
"session": {
|
|
19
|
+
"compute": {
|
|
20
|
+
"defaults": {
|
|
21
|
+
"outputDtype": "f32"
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
},
|
|
25
|
+
"kernelPath": "gemma2-q4k-dequant-f32a-nosubgroups",
|
|
26
|
+
"kernelPathPolicy": {
|
|
27
|
+
"mode": "capability-aware",
|
|
28
|
+
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
29
|
+
"onIncompatible": "remap"
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "kernels/gemma2-q4k-fused-f32a",
|
|
3
|
+
"name": "gemma2-q4k-fused-f32a",
|
|
4
|
+
"description": "Explicit runtime override preset for the Gemma 2 fused Q4K F32 kernel path.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-08T00:00:00Z",
|
|
9
|
+
"extends": "default",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"compute": {
|
|
13
|
+
"activationDtype": "f32"
|
|
14
|
+
},
|
|
15
|
+
"kvcache": {
|
|
16
|
+
"kvDtype": "f32"
|
|
17
|
+
},
|
|
18
|
+
"session": {
|
|
19
|
+
"compute": {
|
|
20
|
+
"defaults": {
|
|
21
|
+
"outputDtype": "f32"
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
},
|
|
25
|
+
"kernelPath": "gemma2-q4k-fused-f32a",
|
|
26
|
+
"kernelPathPolicy": {
|
|
27
|
+
"mode": "capability-aware",
|
|
28
|
+
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
29
|
+
"onIncompatible": "remap"
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
@@ -1,20 +1,13 @@
|
|
|
1
1
|
{
|
|
2
2
|
"id": "kernels/safe-q4k",
|
|
3
3
|
"name": "safe-q4k",
|
|
4
|
-
"description": "
|
|
4
|
+
"description": "Deprecated alias for kernels/gemma2-q4k-dequant-f32a-nosubgroups.",
|
|
5
5
|
"intent": "investigate",
|
|
6
|
-
"stability": "
|
|
6
|
+
"stability": "deprecated",
|
|
7
7
|
"owner": "doppler-core",
|
|
8
8
|
"createdAtUtc": "2026-02-25T00:00:00Z",
|
|
9
|
-
"
|
|
10
|
-
"
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
"kernelPathPolicy": {
|
|
14
|
-
"mode": "capability-aware",
|
|
15
|
-
"sourceScope": ["config", "model", "manifest", "execution-v0"],
|
|
16
|
-
"onIncompatible": "remap"
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
}
|
|
9
|
+
"deprecatedAtUtc": "2026-03-08T00:00:00Z",
|
|
10
|
+
"replacementId": "kernels/gemma2-q4k-dequant-f32a-nosubgroups",
|
|
11
|
+
"extends": "kernels/gemma2-q4k-dequant-f32a-nosubgroups",
|
|
12
|
+
"runtime": {}
|
|
20
13
|
}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "model/qwen3-5-layer-probe",
|
|
3
|
+
"name": "qwen3-5-layer-probe",
|
|
4
|
+
"description": "Probe all 24 layer outputs in Qwen 3.5 to isolate where the hidden state distribution collapses.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-13T00:00:00Z",
|
|
9
|
+
"extends": "modes/debug",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"prompt": "What color is the sky on a clear day? Answer in one word.",
|
|
13
|
+
"batching": {
|
|
14
|
+
"maxTokens": 1
|
|
15
|
+
},
|
|
16
|
+
"sampling": {
|
|
17
|
+
"temperature": 0
|
|
18
|
+
}
|
|
19
|
+
},
|
|
20
|
+
"shared": {
|
|
21
|
+
"debug": {
|
|
22
|
+
"trace": {
|
|
23
|
+
"enabled": true,
|
|
24
|
+
"categories": ["attn", "ffn", "logits"],
|
|
25
|
+
"layers": null,
|
|
26
|
+
"maxDecodeSteps": 1
|
|
27
|
+
},
|
|
28
|
+
"probes": [
|
|
29
|
+
{
|
|
30
|
+
"id": "embed",
|
|
31
|
+
"stage": "embed_out",
|
|
32
|
+
"tokens": [-1],
|
|
33
|
+
"dims": [0, 1, 2, 3, 512, 513]
|
|
34
|
+
},
|
|
35
|
+
{
|
|
36
|
+
"id": "layer_out",
|
|
37
|
+
"stage": "layer_out",
|
|
38
|
+
"layers": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23],
|
|
39
|
+
"tokens": [-1],
|
|
40
|
+
"dims": [0, 1, 2, 3]
|
|
41
|
+
},
|
|
42
|
+
{
|
|
43
|
+
"id": "logits",
|
|
44
|
+
"stage": "logits_final",
|
|
45
|
+
"tokens": [-1],
|
|
46
|
+
"dims": [271, 0, 1, 2, 3, 496, 138]
|
|
47
|
+
}
|
|
48
|
+
]
|
|
49
|
+
}
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
{
|
|
2
|
+
"id": "model/qwen3-5-linear-attn-debug",
|
|
3
|
+
"name": "qwen3-5-linear-attn-debug",
|
|
4
|
+
"description": "Probe linear attention intermediates in Qwen 3.5 layer 0 for comparison with HF reference.",
|
|
5
|
+
"intent": "investigate",
|
|
6
|
+
"stability": "canonical",
|
|
7
|
+
"owner": "doppler-core",
|
|
8
|
+
"createdAtUtc": "2026-03-13T00:00:00Z",
|
|
9
|
+
"extends": "modes/debug",
|
|
10
|
+
"runtime": {
|
|
11
|
+
"inference": {
|
|
12
|
+
"prompt": "Hello",
|
|
13
|
+
"batching": {
|
|
14
|
+
"maxTokens": 1
|
|
15
|
+
},
|
|
16
|
+
"sampling": {
|
|
17
|
+
"temperature": 0
|
|
18
|
+
},
|
|
19
|
+
"chatTemplate": {
|
|
20
|
+
"enabled": false
|
|
21
|
+
}
|
|
22
|
+
},
|
|
23
|
+
"shared": {
|
|
24
|
+
"debug": {
|
|
25
|
+
"trace": {
|
|
26
|
+
"enabled": true,
|
|
27
|
+
"categories": ["attn", "logits"],
|
|
28
|
+
"layers": null,
|
|
29
|
+
"maxDecodeSteps": 1
|
|
30
|
+
},
|
|
31
|
+
"probes": [
|
|
32
|
+
{
|
|
33
|
+
"id": "embed",
|
|
34
|
+
"stage": "embed_out",
|
|
35
|
+
"tokens": [-1],
|
|
36
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
37
|
+
},
|
|
38
|
+
{
|
|
39
|
+
"id": "qkv",
|
|
40
|
+
"stage": "linear_qkv_proj",
|
|
41
|
+
"layers": [0],
|
|
42
|
+
"tokens": [-1],
|
|
43
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
44
|
+
},
|
|
45
|
+
{
|
|
46
|
+
"id": "z",
|
|
47
|
+
"stage": "linear_z_proj",
|
|
48
|
+
"layers": [0],
|
|
49
|
+
"tokens": [-1],
|
|
50
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
51
|
+
},
|
|
52
|
+
{
|
|
53
|
+
"id": "a",
|
|
54
|
+
"stage": "linear_a_proj",
|
|
55
|
+
"layers": [0],
|
|
56
|
+
"tokens": [-1],
|
|
57
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
58
|
+
},
|
|
59
|
+
{
|
|
60
|
+
"id": "b",
|
|
61
|
+
"stage": "linear_b_proj",
|
|
62
|
+
"layers": [0],
|
|
63
|
+
"tokens": [-1],
|
|
64
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
65
|
+
},
|
|
66
|
+
{
|
|
67
|
+
"id": "core",
|
|
68
|
+
"stage": "linear_core_out",
|
|
69
|
+
"layers": [0],
|
|
70
|
+
"tokens": [-1],
|
|
71
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
72
|
+
},
|
|
73
|
+
{
|
|
74
|
+
"id": "layer0",
|
|
75
|
+
"stage": "layer_out",
|
|
76
|
+
"layers": [0],
|
|
77
|
+
"tokens": [-1],
|
|
78
|
+
"dims": [0, 1, 2, 3, 4, 5, 6, 7]
|
|
79
|
+
},
|
|
80
|
+
{
|
|
81
|
+
"id": "logits",
|
|
82
|
+
"stage": "logits_final",
|
|
83
|
+
"tokens": [-1],
|
|
84
|
+
"dims": [0, 1, 2, 3]
|
|
85
|
+
}
|
|
86
|
+
]
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
}
|
package/src/config/runtime.js
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
import { createDopplerConfig, setKernelThresholds } from './schema/index.js';
|
|
2
2
|
import { validateRuntimeConfig, validateRuntimeOverrides } from './param-validator.js';
|
|
3
|
+
import { isPlainObject } from '../utils/plain-object.js';
|
|
3
4
|
|
|
4
5
|
let runtimeConfig = createDopplerConfig().runtime;
|
|
5
6
|
setKernelThresholds(runtimeConfig.shared.kernelThresholds);
|
|
@@ -9,12 +10,16 @@ export function getRuntimeConfig() {
|
|
|
9
10
|
}
|
|
10
11
|
|
|
11
12
|
export function setRuntimeConfig(overrides) {
|
|
12
|
-
if (
|
|
13
|
+
if (overrides === undefined || overrides === null) {
|
|
13
14
|
runtimeConfig = createDopplerConfig().runtime;
|
|
14
15
|
setKernelThresholds(runtimeConfig.shared.kernelThresholds);
|
|
15
16
|
return runtimeConfig;
|
|
16
17
|
}
|
|
17
18
|
|
|
19
|
+
if (!isPlainObject(overrides)) {
|
|
20
|
+
throw new Error('DopplerConfigError: runtime overrides must be an object when provided.');
|
|
21
|
+
}
|
|
22
|
+
|
|
18
23
|
assertNoDeprecatedRuntimeKeys(overrides);
|
|
19
24
|
validateRuntimeOverrides(overrides);
|
|
20
25
|
|
|
@@ -181,6 +181,11 @@ export type ProbeStage =
|
|
|
181
181
|
// Attention stages (per-layer)
|
|
182
182
|
| 'attn_input' // Input to attention (after residual from previous layer)
|
|
183
183
|
| 'attn_normed' // After input RMSNorm
|
|
184
|
+
| 'linear_qkv_proj' // Linear-attention fused QKV projection output
|
|
185
|
+
| 'linear_z_proj' // Linear-attention z projection output
|
|
186
|
+
| 'linear_a_proj' // Linear-attention a projection output
|
|
187
|
+
| 'linear_b_proj' // Linear-attention b projection output
|
|
188
|
+
| 'linear_core_out' // Linear-attention recurrent core output (before o_proj)
|
|
184
189
|
| 'q_proj' // Q projection output
|
|
185
190
|
| 'k_proj' // K projection output
|
|
186
191
|
| 'v_proj' // V projection output
|
|
@@ -5,6 +5,7 @@ import { DEFAULT_EMULATION_CONFIG, createEmulationConfig } from './emulation.sch
|
|
|
5
5
|
import { mergeEcosystemConfig } from './ecosystem.schema.js';
|
|
6
6
|
import {
|
|
7
7
|
chooseNullish,
|
|
8
|
+
chooseDefined,
|
|
8
9
|
mergeExecutionPatchLists,
|
|
9
10
|
mergeKernelPathPolicy,
|
|
10
11
|
mergeShallowObject,
|
|
@@ -31,6 +32,13 @@ export const DEFAULT_DOPPLER_CONFIG = {
|
|
|
31
32
|
runtime: DEFAULT_RUNTIME_CONFIG,
|
|
32
33
|
};
|
|
33
34
|
|
|
35
|
+
function cloneConfigTree(value) {
|
|
36
|
+
if (typeof structuredClone === 'function') {
|
|
37
|
+
return structuredClone(value);
|
|
38
|
+
}
|
|
39
|
+
return JSON.parse(JSON.stringify(value));
|
|
40
|
+
}
|
|
41
|
+
|
|
34
42
|
// =============================================================================
|
|
35
43
|
// Factory Function
|
|
36
44
|
// =============================================================================
|
|
@@ -39,19 +47,21 @@ export function createDopplerConfig(
|
|
|
39
47
|
overrides
|
|
40
48
|
) {
|
|
41
49
|
if (!overrides) {
|
|
42
|
-
return {
|
|
50
|
+
return {
|
|
51
|
+
model: DEFAULT_DOPPLER_CONFIG.model,
|
|
52
|
+
runtime: cloneConfigTree(DEFAULT_RUNTIME_CONFIG),
|
|
53
|
+
};
|
|
43
54
|
}
|
|
44
55
|
|
|
45
56
|
const runtimeOverrides = overrides.runtime ?? {};
|
|
57
|
+
const runtimeBase = cloneConfigTree(DEFAULT_RUNTIME_CONFIG);
|
|
46
58
|
const runtime = overrides.runtime
|
|
47
|
-
? mergeRuntimeConfig(
|
|
48
|
-
:
|
|
59
|
+
? mergeRuntimeConfig(runtimeBase, runtimeOverrides)
|
|
60
|
+
: runtimeBase;
|
|
49
61
|
const config = {
|
|
50
62
|
model: overrides.model ?? DEFAULT_DOPPLER_CONFIG.model,
|
|
51
63
|
runtime,
|
|
52
64
|
};
|
|
53
|
-
|
|
54
|
-
applyCalibrateDefaults(config.runtime, runtimeOverrides);
|
|
55
65
|
return config;
|
|
56
66
|
}
|
|
57
67
|
|
|
@@ -239,7 +249,7 @@ function mergeInferenceConfig(
|
|
|
239
249
|
speculative: { ...base.speculative, ...overrides.speculative },
|
|
240
250
|
generation: { ...base.generation, ...overrides.generation },
|
|
241
251
|
pipeline: overrides.pipeline ?? base.pipeline,
|
|
242
|
-
kernelPath: overrides.kernelPath
|
|
252
|
+
kernelPath: chooseDefined(overrides.kernelPath, base.kernelPath),
|
|
243
253
|
kernelPathSource: overrides.kernelPathSource ?? base.kernelPathSource,
|
|
244
254
|
kernelPathPolicy: mergeKernelPathPolicy(baseKernelPathPolicy, overrideKernelPathPolicy),
|
|
245
255
|
chatTemplate: mergeShallowObject(base.chatTemplate, overrides.chatTemplate),
|
|
@@ -302,21 +312,6 @@ function mergeDebugConfig(
|
|
|
302
312
|
};
|
|
303
313
|
}
|
|
304
314
|
|
|
305
|
-
function applyCalibrateDefaults(runtime, runtimeOverrides) {
|
|
306
|
-
const intent = runtime?.shared?.tooling?.intent;
|
|
307
|
-
if (intent !== 'calibrate') return;
|
|
308
|
-
|
|
309
|
-
const warmupOverrides = runtimeOverrides?.shared?.kernelWarmup;
|
|
310
|
-
const hasPrewarmOverride = warmupOverrides
|
|
311
|
-
&& Object.prototype.hasOwnProperty.call(warmupOverrides, 'prewarm');
|
|
312
|
-
if (!hasPrewarmOverride) {
|
|
313
|
-
runtime.shared.kernelWarmup = {
|
|
314
|
-
...runtime.shared.kernelWarmup,
|
|
315
|
-
prewarm: true,
|
|
316
|
-
};
|
|
317
|
-
}
|
|
318
|
-
}
|
|
319
|
-
|
|
320
315
|
function mergeBenchmarkConfig(
|
|
321
316
|
base,
|
|
322
317
|
overrides
|
|
@@ -172,9 +172,9 @@ export const DEFAULT_PRESET_INFERENCE_CONFIG = {
|
|
|
172
172
|
ropeScalingFactor: 1.0,
|
|
173
173
|
ropeLocalScalingType: null,
|
|
174
174
|
ropeLocalScalingFactor: 1.0,
|
|
175
|
-
yarnBetaFast:
|
|
176
|
-
yarnBetaSlow:
|
|
177
|
-
yarnOriginalMaxPos:
|
|
175
|
+
yarnBetaFast: null,
|
|
176
|
+
yarnBetaSlow: null,
|
|
177
|
+
yarnOriginalMaxPos: null,
|
|
178
178
|
ropeLocalYarnBetaFast: null,
|
|
179
179
|
ropeLocalYarnBetaSlow: null,
|
|
180
180
|
ropeLocalYarnOriginalMaxPos: null,
|
|
@@ -151,13 +151,17 @@ export interface KernelPathSchema {
|
|
|
151
151
|
export type BuiltinKernelPathId =
|
|
152
152
|
| 'gemma2-q4k-fused-f32a' // Gemma 2 Q4K weights, fused matmul, F32 activations
|
|
153
153
|
| 'gemma2-q4k-dequant-f16a' // Gemma 2 Q4K -> F16 dequant, F16 activations
|
|
154
|
+
| 'gemma2-q4k-dequant-f32a-nosubgroups' // Gemma 2 Q4K -> F32 dequant path with no subgroup requirement
|
|
155
|
+
| 'gemma2-q4k-dequant-f32a' // Legacy alias for gemma2-q4k-dequant-f32a-nosubgroups
|
|
154
156
|
| 'gemma2-f16-f16a' // Gemma 2 F16 weights, F16 activations
|
|
155
157
|
| 'gemma2-f16-f32a' // Gemma 2 F16 weights, F32 activations
|
|
156
158
|
| 'gemma3-f16-fused-f16a-online' // Gemma 3 F16 fused FFN online path
|
|
157
159
|
| 'gemma3-f16-fused-f32a-online' // Gemma 3 F16 fused FFN online path with F32 activations
|
|
160
|
+
| 'gemma3-f16-fused-f32a-online-streamingprefill' // Gemma 3 F16 fused FFN online path with streaming prefill attention
|
|
158
161
|
| 'gemma3-q4k-dequant-f16a-online' // Gemma 3 Q4K dequant online path (F16 activations)
|
|
159
162
|
| 'gemma3-q4k-dequant-f32a-online' // Gemma 3 Q4K dequant online path with F32 activations
|
|
160
|
-
| 'gemma3-q4k-dequant-f32a' // Gemma 3 Q4K dequant path with
|
|
163
|
+
| 'gemma3-q4k-dequant-f32a-nosubgroups' // Gemma 3 Q4K dequant path with no subgroup requirement
|
|
164
|
+
| 'gemma3-q4k-dequant-f32a' // Legacy alias for gemma3-q4k-dequant-f32a-nosubgroups
|
|
161
165
|
| 'lfm2-q4k-dequant-f32a-online' // LFM2 Q4K path with F32 activations and fast prefill
|
|
162
166
|
| 'embeddinggemma-f16-f32a' // EmbeddingGemma F16 weights, F32 activations
|
|
163
167
|
| 'embeddinggemma-f32-f32a' // EmbeddingGemma F32 weights, F32 activations
|
|
@@ -155,18 +155,25 @@ export const DEFAULT_KERNEL_THRESHOLDS = {
|
|
|
155
155
|
tuner: DEFAULT_TUNER_LIMITS,
|
|
156
156
|
};
|
|
157
157
|
|
|
158
|
+
function cloneThresholdTree(value) {
|
|
159
|
+
if (typeof structuredClone === 'function') {
|
|
160
|
+
return structuredClone(value);
|
|
161
|
+
}
|
|
162
|
+
return JSON.parse(JSON.stringify(value));
|
|
163
|
+
}
|
|
164
|
+
|
|
158
165
|
// =============================================================================
|
|
159
166
|
// Runtime Access
|
|
160
167
|
// =============================================================================
|
|
161
168
|
|
|
162
|
-
let currentThresholds =
|
|
169
|
+
let currentThresholds = cloneThresholdTree(DEFAULT_KERNEL_THRESHOLDS);
|
|
163
170
|
|
|
164
171
|
export function getKernelThresholds() {
|
|
165
|
-
return currentThresholds;
|
|
172
|
+
return cloneThresholdTree(currentThresholds);
|
|
166
173
|
}
|
|
167
174
|
|
|
168
175
|
export function setKernelThresholds(overrides) {
|
|
169
|
-
|
|
176
|
+
const nextThresholds = {
|
|
170
177
|
...currentThresholds,
|
|
171
178
|
...overrides,
|
|
172
179
|
matmul: { ...currentThresholds.matmul, ...overrides.matmul },
|
|
@@ -180,8 +187,9 @@ export function setKernelThresholds(overrides) {
|
|
|
180
187
|
cast: { ...currentThresholds.cast, ...overrides.cast },
|
|
181
188
|
tuner: { ...currentThresholds.tuner, ...overrides.tuner },
|
|
182
189
|
};
|
|
190
|
+
currentThresholds = cloneThresholdTree(nextThresholds);
|
|
183
191
|
}
|
|
184
192
|
|
|
185
193
|
export function resetKernelThresholds() {
|
|
186
|
-
currentThresholds =
|
|
194
|
+
currentThresholds = cloneThresholdTree(DEFAULT_KERNEL_THRESHOLDS);
|
|
187
195
|
}
|
|
@@ -88,7 +88,8 @@ export interface QuantizationInfoSchema {
|
|
|
88
88
|
tts?: QuantizationValue; // TTS decoder
|
|
89
89
|
projector?: QuantizationValue; // Cross-modal projection layers
|
|
90
90
|
|
|
91
|
-
// Runtime hints
|
|
91
|
+
// Runtime hints. `compute` may be included in variantTag when artifact naming
|
|
92
|
+
// treats activation dtype as part of the published variant identity.
|
|
92
93
|
kvCache?: QuantizationValue;
|
|
93
94
|
compute?: QuantizationValue;
|
|
94
95
|
|
|
@@ -162,7 +163,7 @@ export interface ArchitectureSchema {
|
|
|
162
163
|
* Use `null` to indicate "not applicable" (e.g., no softcapping).
|
|
163
164
|
*/
|
|
164
165
|
export interface ManifestAttentionSchema {
|
|
165
|
-
/** Query pre-attention scalar
|
|
166
|
+
/** Query pre-attention scalar: attnScale = 1/sqrt(scalar). Standard = headDim. */
|
|
166
167
|
queryPreAttnScalar: number;
|
|
167
168
|
/** Attention logit softcapping (Gemma 2: 50, null = disabled) */
|
|
168
169
|
attnLogitSoftcapping: number | null;
|