2 files changed, 41 insertions, 1 deletions
diff --git a/ops/Data/Array/Strided/Arith/Internal.hs b/ops/Data/Array/Strided/Arith/Internal.hs
index 5802573..d94fc65 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
@@ -830,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/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"
   ]