We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9d2c9a7 commit e607a7eCopy full SHA for e607a7e
1 file changed
src/ArrayFire/Orphans.hs
@@ -23,6 +23,7 @@ import Control.DeepSeq (NFData(..))
23
import qualified ArrayFire.Arith as A
24
import qualified ArrayFire.Array as A
25
import qualified ArrayFire.Algorithm as A
26
+import qualified ArrayFire.Data as A
27
import ArrayFire.Types
28
import ArrayFire.Util
29
@@ -39,7 +40,7 @@ instance (Num a, AFType a) => Num (Array a) where
39
40
x + y = A.add x y
41
x * y = A.mul x y
42
abs = A.abs
- signum x = A.cast (A.gt x 0) - A.cast (A.lt x 0)
43
+ signum x = A.select (A.gt x 0) 1 (A.select (A.lt x 0) (-1) 0)
44
negate arr = A.scalar @a (fromInteger (-1)) `A.mul` arr
45
x - y = A.sub x y
46
fromInteger = A.scalar . fromIntegral
0 commit comments