Complex: Add c{double,float}_abs_squared

inducer · Aug 13, 2016 · 96b08c4 · 96b08c4
1 parent 717cf0c
commit 96b08c4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyopencl/cl/pyopencl-complex.h b/pyopencl/cl/pyopencl-complex.h
@@ -35,6 +35,7 @@
   REAL_TP TPROOT##_real(TP a) { return a.real; } \
   REAL_TP TPROOT##_imag(TP a) { return a.imag; } \
   REAL_TP TPROOT##_abs(TP a) { return hypot(a.real, a.imag); } \
+  REAL_TP TPROOT##_abs_squared(TP a) { return a.real * a.real + a.imag * a.imag; } \
   \
   TP TPROOT##_new(REAL_TP real, REAL_TP imag) \
   { \