From e21fef17df4d9a6cdb281e916172dbdf957b3b0a Mon Sep 17 00:00:00 2001 From: Peter Boyle Date: Mon, 31 Aug 2020 23:56:26 -0400 Subject: [PATCH] real and imag part not in ET --- Grid/lattice/Lattice_real_imag.h | 79 ++++++++++++++++++++++++++++++++ 1 file changed, 79 insertions(+) create mode 100644 Grid/lattice/Lattice_real_imag.h diff --git a/Grid/lattice/Lattice_real_imag.h b/Grid/lattice/Lattice_real_imag.h new file mode 100644 index 00000000..003300cc --- /dev/null +++ b/Grid/lattice/Lattice_real_imag.h @@ -0,0 +1,79 @@ +/************************************************************************************* + + Grid physics library, www.github.com/paboyle/Grid + + Source file: ./lib/lattice/Lattice_reality.h + + Copyright (C) 2015 + +Author: Azusa Yamaguchi +Author: Peter Boyle +Author: neo + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License along + with this program; if not, write to the Free Software Foundation, Inc., + 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + + See the full license in the file "LICENSE" in the top level distribution directory +*************************************************************************************/ +/* END LEGAL */ +#ifndef GRID_LATTICE_REAL_IMAG_H +#define GRID_LATTICE_REAL_IMAG_H + + +// FIXME .. this is the sector of the code +// I am most worried about the directions +// The choice of burying complex in the SIMD +// is making the use of "real" and "imag" very cumbersome + +NAMESPACE_BEGIN(Grid); + +template inline Lattice real(const Lattice &lhs){ + Lattice ret(lhs.Grid()); + + autoView( lhs_v, lhs, AcceleratorRead); + autoView( ret_v, ret, AcceleratorWrite); + + ret.Checkerboard()=lhs.Checkerboard(); + accelerator_for( ss, lhs_v.size(), 1, { + ret_v[ss] =real(lhs_v[ss]); + }); + return ret; +}; +template inline Lattice imag(const Lattice &lhs){ + Lattice ret(lhs.Grid()); + + autoView( lhs_v, lhs, AcceleratorRead); + autoView( ret_v, ret, AcceleratorWrite); + + ret.Checkerboard()=lhs.Checkerboard(); + accelerator_for( ss, lhs_v.size(), 1, { + ret_v[ss] =imag(lhs_v[ss]); + }); + return ret; +}; + +template::value,void>::type * = nullptr> + auto real(const Expression &expr) -> decltype(real(closure(expr))) +{ + return real(closure(expr)); +} +template::value,void>::type * = nullptr> + auto imag(const Expression &expr) -> decltype(imag(closure(expr))) +{ + return imag(closure(expr)); +} + +NAMESPACE_END(Grid); + +#endif