From df57cbef77a607a4298fc7e0644aead2392f7c0e Mon Sep 17 00:00:00 2001 From: Jonathan Mulvey Date: Thu, 29 Jan 2026 00:16:24 +0100 Subject: [PATCH] Added support for saving 21x21 clusters (only saving, nothing else yet) --- python/src/module.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python/src/module.cpp b/python/src/module.cpp index f78d2d4..e3fad27 100644 --- a/python/src/module.cpp +++ b/python/src/module.cpp @@ -91,6 +91,10 @@ PYBIND11_MODULE(_aare, m) { DEFINE_CLUSTER_BINDINGS(double, 9, 9, uint16_t, d); DEFINE_CLUSTER_BINDINGS(float, 9, 9, uint16_t, f); + DEFINE_CLUSTER_BINDINGS(int, 21, 21, uint16_t, i); + DEFINE_CLUSTER_BINDINGS(double, 21, 21, uint16_t, d); + DEFINE_CLUSTER_BINDINGS(float, 21, 21, uint16_t, f); + DEFINE_CLUSTER_BINDINGS(int16_t, 3, 3, uint16_t, i16); DEFINE_BINDINGS_CLUSTERFINDER(int, 3, 3, uint16_t, i); @@ -109,6 +113,10 @@ PYBIND11_MODULE(_aare, m) { DEFINE_BINDINGS_CLUSTERFINDER(double, 9, 9, uint16_t, d); DEFINE_BINDINGS_CLUSTERFINDER(float, 9, 9, uint16_t, f); + DEFINE_BINDINGS_CLUSTERFINDER(int, 21, 21, uint16_t, i); + DEFINE_BINDINGS_CLUSTERFINDER(double, 21, 21, uint16_t, d); + DEFINE_BINDINGS_CLUSTERFINDER(float, 21, 21, uint16_t, f); + define_3x3_reduction(m); define_3x3_reduction(m); define_3x3_reduction(m);