Skip to content

Faster Nonlocal denoise: GEMM ONNX graph, halo 40, remove Maximum - #19

Merged
sheldonxxxx merged 1 commit into
mainfrom
feat/nonlocal-halo40-gemm-single-pass
Oct 1, 2026
Merged

sheldonxxxx merged 1 commit into
mainfrom
feat/nonlocal-halo40-gemm-single-pass

Conversation

@sheldonxxxx

Copy link
Copy Markdown
Owner

Summary

  • Rewritten ONNX graph (gemm-1x1-v2): 370 -> 153 ms per 320x320 tile on an RTX 5060 Ti, same frozen fixture gates on CUDA and CPU.
  • Tile halo 64 -> 40: 150 tiles instead of 247 for 32.5 MP (about 23 s model time, estimated).
  • Single accepted model hash; the Hugging Face model repo is republished (nonlocal-v2) and pinned. Older installs are rejected until reinstalled with install_model kind=nonlocal.
  • Removed the Maximum four-rotation option and quality everywhere (breaking, no compatibility shim).
  • Docs, CHANGELOG, skills and model card updated.

Testing

  • cargo fmt --check, cargo clippy --all-targets --all-features -- -D warnings, cargo test --lib (224 passed)
  • Prettier, tsc, eslint, frontend tests (133), MCP suite (147)
  • Python suite (145 passed, 7 skipped) plus checkpoint/CUDA-gated parity tests on the GPU host
  • Ignored install tests for the published ONNX and CoreML bundles pass
  • Not run: full-photo MCP denoise against the installed Hub bundle; CoreML full-photo parity at halo 40

🤖 Generated with Claude Code

…ality

- Derive the ONNX model with gemm-1x1-v2 (drop zero-bias nodes, run 1x1
  convolutions as matrix multiplications): 370 -> 153 ms per tile on CUDA,
  same frozen fixture gates.
- Tile halo 64 -> 40 (150 tiles instead of 247 for 32.5 MP); a single
  accepted model hash with 320/40/240 tiling.
- Remove the Maximum / four-rotation option and the quality parameter from
  the native pipeline, MCP schema, Python tooling, docs and skills.
- Pin the republished Hugging Face revision (nonlocal-v2) and add an ONNX
  install test against it.
- Includes pending MCP session-storage and render/session edits.

Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
@sheldonxxxx
sheldonxxxx merged commit 4494beb into main Oct 1, 2026
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant