From ad3f44c8b170298e63b8b57ee02cb88fbbd210fc Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sat, 15 Nov 2025 11:12:57 +0100
Subject: arith: Support Int8 and Int16

---
 ops/Data/Array/Strided/Arith/Internal.hs       | 30 ++++++++++++++++++++++++++
 ops/Data/Array/Strided/Arith/Internal/Lists.hs |  4 +++-
 2 files changed, 33 insertions(+), 1 deletion(-)

(limited to 'ops')

diff --git a/ops/Data/Array/Strided/Arith/Internal.hs b/ops/Data/Array/Strided/Arith/Internal.hs
index 5802573..6364802 100644
--- a/ops/Data/Array/Strided/Arith/Internal.hs
+++ b/ops/Data/Array/Strided/Arith/Internal.hs
@@ -714,6 +714,36 @@ class NumElt a where
   numEltMaxIndex :: SNat n -> Array n a -> [Int]
   numEltDotprodInner :: SNat n -> Array (n + 1) a -> Array (n + 1) a -> Array n a
 
+instance NumElt Int8 where
+  numEltAdd = addVectorInt8
+  numEltSub = subVectorInt8
+  numEltMul = mulVectorInt8
+  numEltNeg = negVectorInt8
+  numEltAbs = absVectorInt8
+  numEltSignum = signumVectorInt8
+  numEltSum1Inner = sum1VectorInt8
+  numEltProduct1Inner = product1VectorInt8
+  numEltSumFull = sumFullVectorInt8
+  numEltProductFull = productFullVectorInt8
+  numEltMinIndex _ = minindexVectorInt8
+  numEltMaxIndex _ = maxindexVectorInt8
+  numEltDotprodInner = dotprodinnerVectorInt8
+
+instance NumElt Int16 where
+  numEltAdd = addVectorInt16
+  numEltSub = subVectorInt16
+  numEltMul = mulVectorInt16
+  numEltNeg = negVectorInt16
+  numEltAbs = absVectorInt16
+  numEltSignum = signumVectorInt16
+  numEltSum1Inner = sum1VectorInt16
+  numEltProduct1Inner = product1VectorInt16
+  numEltSumFull = sumFullVectorInt16
+  numEltProductFull = productFullVectorInt16
+  numEltMinIndex _ = minindexVectorInt16
+  numEltMaxIndex _ = maxindexVectorInt16
+  numEltDotprodInner = dotprodinnerVectorInt16
+
 instance NumElt Int32 where
   numEltAdd = addVectorInt32
   numEltSub = subVectorInt32
diff --git a/ops/Data/Array/Strided/Arith/Internal/Lists.hs b/ops/Data/Array/Strided/Arith/Internal/Lists.hs
index 910a77c..27204d2 100644
--- a/ops/Data/Array/Strided/Arith/Internal/Lists.hs
+++ b/ops/Data/Array/Strided/Arith/Internal/Lists.hs
@@ -16,7 +16,9 @@ data ArithType = ArithType
 
 intTypesList :: [ArithType]
 intTypesList =
-  [ArithType ''Int32 "i32"
+  [ArithType ''Int8 "i8"
+  ,ArithType ''Int16 "i16"
+  ,ArithType ''Int32 "i32"
   ,ArithType ''Int64 "i64"
   ]
 
-- 
cgit v1.2.3-70-g09d2