Add ability to set real and imaginary components of complex numbers s…

…eparately
inducer · Aug 26, 2015 · a84c2fe · a84c2fe
1 parent 440936b
commit a84c2fe
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pycuda/cuda/pycuda-complex.hpp b/pycuda/cuda/pycuda-complex.hpp
@@ -87,6 +87,10 @@ struct complex {
   value_type real() const { return _M_re; }
   __device__
   value_type imag() const { return _M_im; }
+  __device__
+  value_type &real() { return _M_re; }
+  __device__
+  value_type &imag() { return _M_im; }
 
   // Arithmetic op= operations involving one real argument.
 
@@ -223,6 +227,8 @@ struct _STLP_CLASS_DECLSPEC complex<float> {
   // Element access.
   value_type __device__ real() const { return _M_re; }
   value_type __device__ imag() const { return _M_im; }
+  value_type & __device__ real() { return _M_re; }
+  value_type & __device__ imag() { return _M_im; }
 
   // Arithmetic op= operations involving one real argument.
 
@@ -393,6 +399,10 @@ struct _STLP_CLASS_DECLSPEC complex<double> {
   value_type real() const { return _M_re; }
   __device__
   value_type imag() const { return _M_im; }
+  __device__
+  value_type &real() { return _M_re; }
+  __device__
+  value_type &imag() { return _M_im; }
 
   // Arithmetic op= operations involving one real argument.