From 08e139de6bfeba885cacec1ad5600b85cd0f0947 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Thu, 13 Mar 2025 11:56:17 +0100 Subject: arith: Correct rank arguments to C wrapper functions --- src/Data/Array/Mixed/Internal/Arith.hs | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/Data/Array/Mixed/Internal/Arith.hs b/src/Data/Array/Mixed/Internal/Arith.hs index c940914..9c560d6 100644 --- a/src/Data/Array/Mixed/Internal/Arith.hs +++ b/src/Data/Array/Mixed/Internal/Arith.hs @@ -100,7 +100,7 @@ liftVEltwise2 sn@SNat valconv ptrconv f_ss f_sv f_vs f_vv (Just (_, 1), Just (blockOff, blockSz)) -> -- scalar * dense let arr2' = RS.fromVector [blockSz] (VS.slice blockOff blockSz vec2) - RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinarySV sn valconv ptrconv f_sv (vec1 VS.! offset1) arr2' + RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinarySV (SNat @1) valconv ptrconv f_sv (vec1 VS.! offset1) arr2' in RS.A (RG.A sh1 (OI.T strides2 (offset2 - blockOff) resvec)) (Just (_, 1), Nothing) -> -- scalar * array @@ -108,7 +108,7 @@ liftVEltwise2 sn@SNat valconv ptrconv f_ss f_sv f_vs f_vv (Just (blockOff, blockSz), Just (_, 1)) -> -- dense * scalar let arr1' = RS.fromVector [blockSz] (VS.slice blockOff blockSz vec1) - RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinaryVS sn valconv ptrconv f_vs arr1' (vec2 VS.! offset2) + RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinaryVS (SNat @1) valconv ptrconv f_vs arr1' (vec2 VS.! offset2) in RS.A (RG.A sh1 (OI.T strides1 (offset1 - blockOff) resvec)) (Nothing, Just (_, 1)) -> -- array * scalar @@ -120,7 +120,7 @@ liftVEltwise2 sn@SNat valconv ptrconv f_ss f_sv f_vs f_vv -> -- dense * dense but the strides match let arr1' = RS.fromVector [blockSz1] (VS.slice blockOff1 blockSz1 vec1) arr2' = RS.fromVector [blockSz1] (VS.slice blockOff2 blockSz2 vec2) - RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinaryVV sn ptrconv f_vv arr1' arr2' + RS.A (RG.A _ (OI.T _ _ resvec)) = wrapBinaryVV (SNat @1) ptrconv f_vv arr1' arr2' in RS.A (RG.A sh1 (OI.T strides1 (offset1 - blockOff1) resvec)) (_, _) -> -- fallback case -- cgit v1.2.3-70-g09d2