add abs() function for float3s

1 files changed, 6 insertions, 0 deletions

diff --git a/chroma/cuda/linalg.h b/chroma/cuda/linalg.h
index 35b2423..6317576 100644
--- a/chroma/cuda/linalg.h
+++ b/chroma/cuda/linalg.h
@@ -155,6 +155,12 @@ cross(const float3 &a, const float3 &b)
     return make_float3(a.y*b.z-a.z*b.y, a.z*b.x-a.x*b.z, a.x*b.y-a.y*b.x);
 }
 
+__device__ float3
+abs(const float3&a)
+{
+    return make_float3(abs(a.x),abs(a.y),abs(a.z));
+}
+
 __device__ float
 norm(const float3 &a)
 {