Skip to content

[3.3] GPU dictionary indexing — the reason this wave exists #71

Description

@CSSFrancis

Normalized cross-correlation between P experimental and D dictionary patterns is, once both are zero-mean / unit-norm, a single matmul E @ Dᵀ(P, D), followed by top-k.

  • Chunk over both P and D with a running top-k so the (P, D) intermediate never materialises — 65k × 100k float32 is 26 GB whole.
  • kikuchipy does this with dask; torch should be dramatically faster.
  • Support NormalizedCrossCorrelationMetric and NormalizedDotProductMetric semantics.

Acceptance: score and index parity against kikuchipy.indexing on the same dictionary, plus a benchmark at real scan scale.

Part of #68.


Plan: RELEASE_0_3_0_PLAN.md

Metadata

Metadata

Assignees

No one assigned

    Labels

    0.3.0SpyDE 0.3.0 releasewave-3Wave 3

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions