Enable setting of real and imaginary parts using C++11 interface inst…

…ead of returning by reference.
inducer · Aug 27, 2015 · 4e25f3f · 4e25f3f
1 parent a84c2fe
commit 4e25f3f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/pycuda/cuda/pycuda-complex.hpp b/pycuda/cuda/pycuda-complex.hpp
@@ -88,9 +88,9 @@ struct complex {
   __device__
   value_type imag() const { return _M_im; }
   __device__
-  value_type &real() { return _M_re; }
+  void real(value_type val) { _M_re = val; }
   __device__
-  value_type &imag() { return _M_im; }
+  void imag(value_type val) { _M_im = val; }
 
   // Arithmetic op= operations involving one real argument.
 
@@ -227,8 +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; }
+  void __device__ real(value_type val) { _M_re = val; }
+  void __device__ imag(value_type val) { _M_im = val; }
 
   // Arithmetic op= operations involving one real argument.
 
@@ -400,9 +400,9 @@ struct _STLP_CLASS_DECLSPEC complex<double> {
   __device__
   value_type imag() const { return _M_im; }
   __device__
-  value_type &real() { return _M_re; }
+  void real(value_type val) { _M_re = val; }
   __device__
-  value_type &imag() { return _M_im; }
+  void imag(value_type val) { _M_im = val; }
 
   // Arithmetic op= operations involving one real argument.