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 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

(limited to 'ops/Data/Array/Strided/Arith/Internal.hs')

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
-- 
cgit v1.2.3-70-g09d2