@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
|
@@ -19,6 +19,18 @@ function fillRandom(data, rng) {
|
|
|
19
19
|
for (let i = 0; i < data.length; i++) data[i] = rng();
|
|
20
20
|
}
|
|
21
21
|
|
|
22
|
+
function destroyBuffer(buffer) {
|
|
23
|
+
if (buffer && typeof buffer.destroy === 'function') {
|
|
24
|
+
buffer.destroy();
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
|
|
28
|
+
function destroyBuffers(...buffers) {
|
|
29
|
+
for (const buffer of buffers) {
|
|
30
|
+
destroyBuffer(buffer);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
|
|
22
34
|
|
|
23
35
|
export async function benchmarkPipeline(
|
|
24
36
|
device,
|
|
@@ -98,7 +110,6 @@ export async function tuneMatmul(
|
|
|
98
110
|
deviceInfo: capabilities?.adapterInfo,
|
|
99
111
|
};
|
|
100
112
|
|
|
101
|
-
// Create test buffers
|
|
102
113
|
const bufferA = device.createBuffer({
|
|
103
114
|
size: M * K * 4,
|
|
104
115
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST,
|
|
@@ -112,93 +123,86 @@ export async function tuneMatmul(
|
|
|
112
123
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
|
|
113
124
|
});
|
|
114
125
|
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
device.queue.submit([encoder.finish()]);
|
|
157
|
-
}
|
|
158
|
-
await device.queue.onSubmittedWorkDone();
|
|
159
|
-
|
|
160
|
-
// Benchmark
|
|
161
|
-
|
|
162
|
-
const times = [];
|
|
163
|
-
for (let i = 0; i < iterations; i++) {
|
|
164
|
-
const start = performance.now();
|
|
165
|
-
const encoder = device.createCommandEncoder();
|
|
166
|
-
const pass = encoder.beginComputePass();
|
|
167
|
-
pass.setPipeline(pipeline);
|
|
168
|
-
pass.setBindGroup(0, bindGroup);
|
|
169
|
-
pass.dispatchWorkgroups(Math.ceil(M / wgX), Math.ceil(N / wgY));
|
|
170
|
-
pass.end();
|
|
171
|
-
device.queue.submit([encoder.finish()]);
|
|
126
|
+
try {
|
|
127
|
+
const dataA = new Float32Array(M * K);
|
|
128
|
+
const dataB = new Float32Array(K * N);
|
|
129
|
+
const matmulRng = createRng(0x13579bdf);
|
|
130
|
+
fillRandom(dataA, matmulRng);
|
|
131
|
+
fillRandom(dataB, matmulRng);
|
|
132
|
+
device.queue.writeBuffer(bufferA, 0, dataA);
|
|
133
|
+
device.queue.writeBuffer(bufferB, 0, dataB);
|
|
134
|
+
|
|
135
|
+
for (const [wgX, wgY] of matmulCandidates) {
|
|
136
|
+
let uniformBuffer = null;
|
|
137
|
+
try {
|
|
138
|
+
const shader = createMatmulShader(wgX, wgY);
|
|
139
|
+
const pipeline = await createComputePipeline(device, shader, 'main');
|
|
140
|
+
|
|
141
|
+
uniformBuffer = device.createBuffer({
|
|
142
|
+
size: 16,
|
|
143
|
+
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
|
|
144
|
+
});
|
|
145
|
+
const uniformData = new Uint32Array([M, N, K, 0]);
|
|
146
|
+
device.queue.writeBuffer(uniformBuffer, 0, uniformData);
|
|
147
|
+
|
|
148
|
+
const bindGroup = device.createBindGroup({
|
|
149
|
+
layout: pipeline.getBindGroupLayout(0),
|
|
150
|
+
entries: [
|
|
151
|
+
{ binding: 0, resource: { buffer: uniformBuffer } },
|
|
152
|
+
{ binding: 1, resource: { buffer: bufferA } },
|
|
153
|
+
{ binding: 2, resource: { buffer: bufferB } },
|
|
154
|
+
{ binding: 3, resource: { buffer: bufferC } },
|
|
155
|
+
],
|
|
156
|
+
});
|
|
157
|
+
|
|
158
|
+
for (let i = 0; i < warmup; i++) {
|
|
159
|
+
const encoder = device.createCommandEncoder();
|
|
160
|
+
const pass = encoder.beginComputePass();
|
|
161
|
+
pass.setPipeline(pipeline);
|
|
162
|
+
pass.setBindGroup(0, bindGroup);
|
|
163
|
+
pass.dispatchWorkgroups(Math.ceil(M / wgX), Math.ceil(N / wgY));
|
|
164
|
+
pass.end();
|
|
165
|
+
device.queue.submit([encoder.finish()]);
|
|
166
|
+
}
|
|
172
167
|
await device.queue.onSubmittedWorkDone();
|
|
173
|
-
times.push(performance.now() - start);
|
|
174
|
-
}
|
|
175
168
|
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
169
|
+
const times = [];
|
|
170
|
+
for (let i = 0; i < iterations; i++) {
|
|
171
|
+
const start = performance.now();
|
|
172
|
+
const encoder = device.createCommandEncoder();
|
|
173
|
+
const pass = encoder.beginComputePass();
|
|
174
|
+
pass.setPipeline(pipeline);
|
|
175
|
+
pass.setBindGroup(0, bindGroup);
|
|
176
|
+
pass.dispatchWorkgroups(Math.ceil(M / wgX), Math.ceil(N / wgY));
|
|
177
|
+
pass.end();
|
|
178
|
+
device.queue.submit([encoder.finish()]);
|
|
179
|
+
await device.queue.onSubmittedWorkDone();
|
|
180
|
+
times.push(performance.now() - start);
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
const avgTime = times.reduce((a, b) => a + b, 0) / times.length;
|
|
184
|
+
const flops = 2 * M * N * K;
|
|
185
|
+
const gflops = (flops / avgTime) / 1e6;
|
|
186
|
+
|
|
187
|
+
if (avgTime < best.timeMs) {
|
|
188
|
+
best = {
|
|
189
|
+
optimalWorkgroupSize: [wgX, wgY, 1],
|
|
190
|
+
optimalTileSize: wgX,
|
|
191
|
+
throughput: gflops,
|
|
192
|
+
timeMs: avgTime,
|
|
193
|
+
deviceInfo: capabilities?.adapterInfo,
|
|
194
|
+
};
|
|
195
|
+
}
|
|
196
|
+
} catch (e) {
|
|
197
|
+
continue;
|
|
198
|
+
} finally {
|
|
199
|
+
destroyBuffer(uniformBuffer);
|
|
188
200
|
}
|
|
189
|
-
|
|
190
|
-
uniformBuffer.destroy();
|
|
191
|
-
} catch (e) {
|
|
192
|
-
// Skip invalid configurations
|
|
193
|
-
continue;
|
|
194
201
|
}
|
|
202
|
+
} finally {
|
|
203
|
+
destroyBuffers(bufferA, bufferB, bufferC);
|
|
195
204
|
}
|
|
196
205
|
|
|
197
|
-
// Cleanup
|
|
198
|
-
bufferA.destroy();
|
|
199
|
-
bufferB.destroy();
|
|
200
|
-
bufferC.destroy();
|
|
201
|
-
|
|
202
206
|
return best;
|
|
203
207
|
}
|
|
204
208
|
|
|
@@ -277,68 +281,69 @@ export async function tuneAttention(
|
|
|
277
281
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
|
|
278
282
|
});
|
|
279
283
|
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
284
|
+
try {
|
|
285
|
+
const dataQ = new Float32Array(totalElements);
|
|
286
|
+
const dataK = new Float32Array(totalElements);
|
|
287
|
+
const attentionRng = createRng(0x2468ace1);
|
|
288
|
+
fillRandom(dataQ, attentionRng);
|
|
289
|
+
fillRandom(dataK, attentionRng);
|
|
290
|
+
device.queue.writeBuffer(bufferQ, 0, dataQ);
|
|
291
|
+
device.queue.writeBuffer(bufferK, 0, dataK);
|
|
292
|
+
|
|
293
|
+
for (const [wgX] of attentionCandidates) {
|
|
294
|
+
let uniformBuffer = null;
|
|
295
|
+
try {
|
|
296
|
+
const shader = createAttentionShader(wgX);
|
|
297
|
+
const pipeline = await createComputePipeline(device, shader, 'main');
|
|
298
|
+
|
|
299
|
+
uniformBuffer = device.createBuffer({
|
|
300
|
+
size: 16,
|
|
301
|
+
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
|
|
302
|
+
});
|
|
303
|
+
const uniformData = new Uint32Array([headDim, numHeads, benchSeqLen, 0]);
|
|
304
|
+
device.queue.writeBuffer(uniformBuffer, 0, uniformData);
|
|
305
|
+
|
|
306
|
+
const bindGroup = device.createBindGroup({
|
|
307
|
+
layout: pipeline.getBindGroupLayout(0),
|
|
308
|
+
entries: [
|
|
309
|
+
{ binding: 0, resource: { buffer: uniformBuffer } },
|
|
310
|
+
{ binding: 1, resource: { buffer: bufferQ } },
|
|
311
|
+
{ binding: 2, resource: { buffer: bufferK } },
|
|
312
|
+
{ binding: 3, resource: { buffer: bufferOut } },
|
|
313
|
+
],
|
|
314
|
+
});
|
|
315
|
+
|
|
316
|
+
const avgTime = await benchmarkPipeline(
|
|
317
|
+
device,
|
|
318
|
+
pipeline,
|
|
319
|
+
bindGroup,
|
|
320
|
+
[totalHeads, 1, 1],
|
|
321
|
+
warmup,
|
|
322
|
+
iterations
|
|
323
|
+
);
|
|
324
|
+
|
|
325
|
+
const flops = 2 * totalHeads * headDim;
|
|
326
|
+
const gflops = avgTime > 0 ? (flops / avgTime) / 1e6 : 0;
|
|
327
|
+
|
|
328
|
+
if (avgTime < best.timeMs) {
|
|
329
|
+
best = {
|
|
330
|
+
optimalWorkgroupSize: [wgX, 1, 1],
|
|
331
|
+
optimalTileSize: wgX,
|
|
332
|
+
throughput: gflops,
|
|
333
|
+
timeMs: avgTime,
|
|
334
|
+
deviceInfo: capabilities?.adapterInfo,
|
|
335
|
+
};
|
|
336
|
+
}
|
|
337
|
+
} catch (e) {
|
|
338
|
+
continue;
|
|
339
|
+
} finally {
|
|
340
|
+
destroyBuffer(uniformBuffer);
|
|
330
341
|
}
|
|
331
|
-
|
|
332
|
-
uniformBuffer.destroy();
|
|
333
|
-
} catch (e) {
|
|
334
|
-
continue;
|
|
335
342
|
}
|
|
343
|
+
} finally {
|
|
344
|
+
destroyBuffers(bufferQ, bufferK, bufferOut);
|
|
336
345
|
}
|
|
337
346
|
|
|
338
|
-
bufferQ.destroy();
|
|
339
|
-
bufferK.destroy();
|
|
340
|
-
bufferOut.destroy();
|
|
341
|
-
|
|
342
347
|
return best;
|
|
343
348
|
}
|
|
344
349
|
|
|
@@ -437,63 +442,65 @@ export async function tuneSoftmax(
|
|
|
437
442
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
|
|
438
443
|
});
|
|
439
444
|
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
449
|
-
|
|
450
|
-
|
|
451
|
-
|
|
452
|
-
|
|
453
|
-
|
|
454
|
-
|
|
455
|
-
|
|
456
|
-
|
|
457
|
-
|
|
458
|
-
|
|
459
|
-
|
|
460
|
-
|
|
461
|
-
|
|
462
|
-
|
|
463
|
-
|
|
464
|
-
|
|
465
|
-
|
|
466
|
-
|
|
467
|
-
|
|
468
|
-
|
|
469
|
-
|
|
470
|
-
|
|
471
|
-
|
|
472
|
-
|
|
473
|
-
|
|
474
|
-
|
|
475
|
-
|
|
476
|
-
|
|
477
|
-
|
|
478
|
-
|
|
479
|
-
|
|
480
|
-
|
|
481
|
-
|
|
482
|
-
|
|
483
|
-
|
|
484
|
-
|
|
485
|
-
|
|
445
|
+
try {
|
|
446
|
+
const dataIn = new Float32Array(totalElements);
|
|
447
|
+
const softmaxRng = createRng(0x31415926);
|
|
448
|
+
fillRandom(dataIn, softmaxRng);
|
|
449
|
+
device.queue.writeBuffer(bufferIn, 0, dataIn);
|
|
450
|
+
|
|
451
|
+
for (const [wgX] of softmaxCandidates) {
|
|
452
|
+
let uniformBuffer = null;
|
|
453
|
+
try {
|
|
454
|
+
const shader = createSoftmaxShader(wgX);
|
|
455
|
+
const pipeline = await createComputePipeline(device, shader, 'main');
|
|
456
|
+
|
|
457
|
+
uniformBuffer = device.createBuffer({
|
|
458
|
+
size: 16,
|
|
459
|
+
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
|
|
460
|
+
});
|
|
461
|
+
const uniformData = new Uint32Array([innerSize, outerSize, 0, 0]);
|
|
462
|
+
device.queue.writeBuffer(uniformBuffer, 0, uniformData);
|
|
463
|
+
|
|
464
|
+
const bindGroup = device.createBindGroup({
|
|
465
|
+
layout: pipeline.getBindGroupLayout(0),
|
|
466
|
+
entries: [
|
|
467
|
+
{ binding: 0, resource: { buffer: uniformBuffer } },
|
|
468
|
+
{ binding: 1, resource: { buffer: bufferIn } },
|
|
469
|
+
{ binding: 2, resource: { buffer: bufferOut } },
|
|
470
|
+
],
|
|
471
|
+
});
|
|
472
|
+
|
|
473
|
+
const avgTime = await benchmarkPipeline(
|
|
474
|
+
device,
|
|
475
|
+
pipeline,
|
|
476
|
+
bindGroup,
|
|
477
|
+
[outerSize, 1, 1],
|
|
478
|
+
warmup,
|
|
479
|
+
iterations
|
|
480
|
+
);
|
|
481
|
+
|
|
482
|
+
const ops = 2 * totalElements;
|
|
483
|
+
const gops = avgTime > 0 ? (ops / avgTime) / 1e6 : 0;
|
|
484
|
+
|
|
485
|
+
if (avgTime < best.timeMs) {
|
|
486
|
+
best = {
|
|
487
|
+
optimalWorkgroupSize: [wgX, 1, 1],
|
|
488
|
+
optimalTileSize: wgX,
|
|
489
|
+
throughput: gops,
|
|
490
|
+
timeMs: avgTime,
|
|
491
|
+
deviceInfo: capabilities?.adapterInfo,
|
|
492
|
+
};
|
|
493
|
+
}
|
|
494
|
+
} catch (e) {
|
|
495
|
+
continue;
|
|
496
|
+
} finally {
|
|
497
|
+
destroyBuffer(uniformBuffer);
|
|
486
498
|
}
|
|
487
|
-
|
|
488
|
-
uniformBuffer.destroy();
|
|
489
|
-
} catch (e) {
|
|
490
|
-
continue;
|
|
491
499
|
}
|
|
500
|
+
} finally {
|
|
501
|
+
destroyBuffers(bufferIn, bufferOut);
|
|
492
502
|
}
|
|
493
503
|
|
|
494
|
-
bufferIn.destroy();
|
|
495
|
-
bufferOut.destroy();
|
|
496
|
-
|
|
497
504
|
return best;
|
|
498
505
|
}
|
|
499
506
|
|
|
@@ -620,73 +627,74 @@ export async function tuneRMSNorm(
|
|
|
620
627
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
|
|
621
628
|
});
|
|
622
629
|
|
|
623
|
-
|
|
624
|
-
|
|
625
|
-
|
|
626
|
-
|
|
627
|
-
|
|
628
|
-
|
|
629
|
-
|
|
630
|
-
|
|
631
|
-
|
|
632
|
-
|
|
633
|
-
|
|
634
|
-
|
|
635
|
-
|
|
636
|
-
|
|
637
|
-
|
|
638
|
-
|
|
639
|
-
|
|
640
|
-
|
|
641
|
-
|
|
642
|
-
|
|
643
|
-
|
|
644
|
-
|
|
645
|
-
|
|
646
|
-
|
|
647
|
-
|
|
648
|
-
|
|
649
|
-
|
|
650
|
-
|
|
651
|
-
|
|
652
|
-
|
|
653
|
-
|
|
654
|
-
|
|
655
|
-
|
|
656
|
-
|
|
657
|
-
|
|
658
|
-
|
|
659
|
-
|
|
660
|
-
|
|
661
|
-
|
|
662
|
-
|
|
663
|
-
|
|
664
|
-
|
|
665
|
-
|
|
666
|
-
|
|
667
|
-
|
|
668
|
-
|
|
669
|
-
|
|
670
|
-
|
|
671
|
-
|
|
672
|
-
|
|
673
|
-
|
|
674
|
-
|
|
675
|
-
|
|
676
|
-
|
|
677
|
-
|
|
630
|
+
try {
|
|
631
|
+
const dataIn = new Float32Array(totalElements);
|
|
632
|
+
const dataWeight = new Float32Array(hiddenSize);
|
|
633
|
+
const rmsRng = createRng(0x27182818);
|
|
634
|
+
fillRandom(dataIn, rmsRng);
|
|
635
|
+
fillRandom(dataWeight, rmsRng);
|
|
636
|
+
device.queue.writeBuffer(bufferIn, 0, dataIn);
|
|
637
|
+
device.queue.writeBuffer(bufferWeight, 0, dataWeight);
|
|
638
|
+
|
|
639
|
+
for (const [wgX] of rmsCandidates) {
|
|
640
|
+
let uniformBuffer = null;
|
|
641
|
+
try {
|
|
642
|
+
const shader = createRMSNormShader(wgX);
|
|
643
|
+
const pipeline = await createComputePipeline(device, shader, 'main');
|
|
644
|
+
|
|
645
|
+
uniformBuffer = device.createBuffer({
|
|
646
|
+
size: 16,
|
|
647
|
+
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
|
|
648
|
+
});
|
|
649
|
+
const uniformData = new ArrayBuffer(16);
|
|
650
|
+
const uniformView = new DataView(uniformData);
|
|
651
|
+
uniformView.setUint32(0, hiddenSize, true);
|
|
652
|
+
uniformView.setUint32(4, numTokens, true);
|
|
653
|
+
uniformView.setFloat32(8, DEFAULT_RMS_NORM_EPS, true);
|
|
654
|
+
uniformView.setUint32(12, 0, true);
|
|
655
|
+
device.queue.writeBuffer(uniformBuffer, 0, uniformData);
|
|
656
|
+
|
|
657
|
+
const bindGroup = device.createBindGroup({
|
|
658
|
+
layout: pipeline.getBindGroupLayout(0),
|
|
659
|
+
entries: [
|
|
660
|
+
{ binding: 0, resource: { buffer: uniformBuffer } },
|
|
661
|
+
{ binding: 1, resource: { buffer: bufferIn } },
|
|
662
|
+
{ binding: 2, resource: { buffer: bufferWeight } },
|
|
663
|
+
{ binding: 3, resource: { buffer: bufferOut } },
|
|
664
|
+
],
|
|
665
|
+
});
|
|
666
|
+
|
|
667
|
+
const avgTime = await benchmarkPipeline(
|
|
668
|
+
device,
|
|
669
|
+
pipeline,
|
|
670
|
+
bindGroup,
|
|
671
|
+
[numTokens, 1, 1],
|
|
672
|
+
warmup,
|
|
673
|
+
iterations
|
|
674
|
+
);
|
|
675
|
+
|
|
676
|
+
const ops = 2 * totalElements;
|
|
677
|
+
const gops = avgTime > 0 ? (ops / avgTime) / 1e6 : 0;
|
|
678
|
+
|
|
679
|
+
if (avgTime < best.timeMs) {
|
|
680
|
+
best = {
|
|
681
|
+
optimalWorkgroupSize: [wgX, 1, 1],
|
|
682
|
+
optimalTileSize: wgX,
|
|
683
|
+
throughput: gops,
|
|
684
|
+
timeMs: avgTime,
|
|
685
|
+
deviceInfo: capabilities?.adapterInfo,
|
|
686
|
+
};
|
|
687
|
+
}
|
|
688
|
+
} catch (e) {
|
|
689
|
+
continue;
|
|
690
|
+
} finally {
|
|
691
|
+
destroyBuffer(uniformBuffer);
|
|
678
692
|
}
|
|
679
|
-
|
|
680
|
-
uniformBuffer.destroy();
|
|
681
|
-
} catch (e) {
|
|
682
|
-
continue;
|
|
683
693
|
}
|
|
694
|
+
} finally {
|
|
695
|
+
destroyBuffers(bufferIn, bufferWeight, bufferOut);
|
|
684
696
|
}
|
|
685
697
|
|
|
686
|
-
bufferIn.destroy();
|
|
687
|
-
bufferWeight.destroy();
|
|
688
|
-
bufferOut.destroy();
|
|
689
|
-
|
|
690
698
|
return best;
|
|
691
699
|
}
|
|
692
700
|
|
|
@@ -789,70 +797,72 @@ export async function tuneDequant(
|
|
|
789
797
|
usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
|
|
790
798
|
});
|
|
791
799
|
|
|
792
|
-
|
|
793
|
-
|
|
794
|
-
|
|
795
|
-
|
|
796
|
-
|
|
797
|
-
|
|
798
|
-
|
|
799
|
-
|
|
800
|
-
|
|
801
|
-
|
|
802
|
-
|
|
803
|
-
|
|
804
|
-
|
|
805
|
-
|
|
806
|
-
|
|
807
|
-
|
|
808
|
-
|
|
809
|
-
|
|
810
|
-
|
|
811
|
-
|
|
812
|
-
|
|
813
|
-
|
|
814
|
-
|
|
815
|
-
|
|
816
|
-
|
|
817
|
-
|
|
818
|
-
|
|
819
|
-
|
|
820
|
-
|
|
821
|
-
|
|
822
|
-
|
|
823
|
-
|
|
824
|
-
|
|
825
|
-
|
|
826
|
-
|
|
827
|
-
|
|
828
|
-
|
|
829
|
-
|
|
830
|
-
|
|
831
|
-
|
|
832
|
-
|
|
833
|
-
|
|
834
|
-
|
|
835
|
-
|
|
836
|
-
|
|
837
|
-
|
|
838
|
-
|
|
839
|
-
|
|
840
|
-
|
|
841
|
-
|
|
842
|
-
|
|
843
|
-
|
|
844
|
-
|
|
800
|
+
try {
|
|
801
|
+
const dataIn = new Uint32Array(numElements);
|
|
802
|
+
for (let i = 0; i < numElements; i++) {
|
|
803
|
+
dataIn[i] = i & 0xffff;
|
|
804
|
+
}
|
|
805
|
+
device.queue.writeBuffer(bufferIn, 0, dataIn);
|
|
806
|
+
|
|
807
|
+
for (const [wgX] of dequantCandidates) {
|
|
808
|
+
let uniformBuffer = null;
|
|
809
|
+
try {
|
|
810
|
+
const shader = createDequantShader(wgX);
|
|
811
|
+
const pipeline = await createComputePipeline(device, shader, 'main');
|
|
812
|
+
|
|
813
|
+
uniformBuffer = device.createBuffer({
|
|
814
|
+
size: 16,
|
|
815
|
+
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
|
|
816
|
+
});
|
|
817
|
+
const uniformData = new ArrayBuffer(16);
|
|
818
|
+
const uniformView = new DataView(uniformData);
|
|
819
|
+
uniformView.setUint32(0, numElements, true);
|
|
820
|
+
uniformView.setFloat32(4, 0.01, true);
|
|
821
|
+
uniformView.setUint32(8, 0, true);
|
|
822
|
+
uniformView.setUint32(12, 0, true);
|
|
823
|
+
device.queue.writeBuffer(uniformBuffer, 0, uniformData);
|
|
824
|
+
|
|
825
|
+
const bindGroup = device.createBindGroup({
|
|
826
|
+
layout: pipeline.getBindGroupLayout(0),
|
|
827
|
+
entries: [
|
|
828
|
+
{ binding: 0, resource: { buffer: uniformBuffer } },
|
|
829
|
+
{ binding: 1, resource: { buffer: bufferIn } },
|
|
830
|
+
{ binding: 2, resource: { buffer: bufferOut } },
|
|
831
|
+
],
|
|
832
|
+
});
|
|
833
|
+
|
|
834
|
+
const workgroups = Math.ceil(numElements / wgX);
|
|
835
|
+
const avgTime = await benchmarkPipeline(
|
|
836
|
+
device,
|
|
837
|
+
pipeline,
|
|
838
|
+
bindGroup,
|
|
839
|
+
[workgroups, 1, 1],
|
|
840
|
+
warmup,
|
|
841
|
+
iterations
|
|
842
|
+
);
|
|
843
|
+
|
|
844
|
+
const ops = numElements;
|
|
845
|
+
const gops = avgTime > 0 ? (ops / avgTime) / 1e6 : 0;
|
|
846
|
+
|
|
847
|
+
if (avgTime < best.timeMs) {
|
|
848
|
+
best = {
|
|
849
|
+
optimalWorkgroupSize: [wgX, 1, 1],
|
|
850
|
+
optimalTileSize: wgX,
|
|
851
|
+
throughput: gops,
|
|
852
|
+
timeMs: avgTime,
|
|
853
|
+
deviceInfo: capabilities?.adapterInfo,
|
|
854
|
+
};
|
|
855
|
+
}
|
|
856
|
+
} catch (e) {
|
|
857
|
+
continue;
|
|
858
|
+
} finally {
|
|
859
|
+
destroyBuffer(uniformBuffer);
|
|
845
860
|
}
|
|
846
|
-
|
|
847
|
-
uniformBuffer.destroy();
|
|
848
|
-
} catch (e) {
|
|
849
|
-
continue;
|
|
850
861
|
}
|
|
862
|
+
} finally {
|
|
863
|
+
destroyBuffers(bufferIn, bufferOut);
|
|
851
864
|
}
|
|
852
865
|
|
|
853
|
-
bufferIn.destroy();
|
|
854
|
-
bufferOut.destroy();
|
|
855
|
-
|
|
856
866
|
return best;
|
|
857
867
|
}
|
|
858
868
|
|