From f7025bc3727e5ee058df3dfebb1fb73b933ee6b0 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sat, 15 Nov 2025 11:24:00 +0100
Subject: arith: Add IntElt instances for Int8 and Int16

---
 ops/Data/Array/Strided/Arith/Internal.hs | 8 ++++++++
 src/Data/Array/Nested/Mixed.hs           | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/ops/Data/Array/Strided/Arith/Internal.hs b/ops/Data/Array/Strided/Arith/Internal.hs
index 6364802..d94fc65 100644
--- a/ops/Data/Array/Strided/Arith/Internal.hs
+++ b/ops/Data/Array/Strided/Arith/Internal.hs
@@ -860,6 +860,14 @@ class NumElt a => IntElt a where
   intEltQuot :: SNat n -> Array n a -> Array n a -> Array n a
   intEltRem :: SNat n -> Array n a -> Array n a -> Array n a
 
+instance IntElt Int8 where
+  intEltQuot = quotVectorInt8
+  intEltRem = remVectorInt8
+
+instance IntElt Int16 where
+  intEltQuot = quotVectorInt16
+  intEltRem = remVectorInt16
+
 instance IntElt Int32 where
   intEltQuot = quotVectorInt32
   intEltRem = remVectorInt32
diff --git a/src/Data/Array/Nested/Mixed.hs b/src/Data/Array/Nested/Mixed.hs
index 0cdcc42..7bda08c 100644
--- a/src/Data/Array/Nested/Mixed.hs
+++ b/src/Data/Array/Nested/Mixed.hs
@@ -91,6 +91,9 @@ import Data.Bag
 -- Unfortunately, the setup of the library requires us to list these primitive
 -- element types multiple times; to aid in extending the list, all these lists
 -- have been marked with [PRIMITIVE ELEMENT TYPES LIST].
+--
+-- NOTE: if you add PRIMITIVE types, be sure to also add NumElt and IntElt
+-- instances for them!
 
 
 -- | Wrapper type used as a tag to attach instances on. The instances on arrays
-- 
cgit v1.2.3-70-g09d2