mirror of
https://github.com/paboyle/Grid.git
synced 2025-06-22 17:52:02 +01:00
Compare commits
8 Commits
bae0f8ea99
...
feature/fe
Author | SHA1 | Date | |
---|---|---|---|
992431cb41 | |||
f8c70545a0 | |||
4a359fa9e9 | |||
f81b15014a | |||
386e63aeb9 | |||
195b0682e7 | |||
8244b8ee10 | |||
0f706c10ec |
@ -45,7 +45,7 @@ directory
|
|||||||
//disables nvcc specific warning in json.hpp
|
//disables nvcc specific warning in json.hpp
|
||||||
#pragma clang diagnostic ignored "-Wdeprecated-register"
|
#pragma clang diagnostic ignored "-Wdeprecated-register"
|
||||||
|
|
||||||
#ifdef __NVCC_DIAG_PRAGMA_SUPPORT__
|
#if (__CUDACC_VER_MAJOR__ >= 11) && (__CUDACC_VER_MINOR__ >= 5)
|
||||||
//disables nvcc specific warning in json.hpp
|
//disables nvcc specific warning in json.hpp
|
||||||
#pragma nv_diag_suppress unsigned_compare_with_zero
|
#pragma nv_diag_suppress unsigned_compare_with_zero
|
||||||
#pragma nv_diag_suppress cast_to_qualified_type
|
#pragma nv_diag_suppress cast_to_qualified_type
|
||||||
|
@ -14,7 +14,7 @@
|
|||||||
/* NVCC save and restore compile environment*/
|
/* NVCC save and restore compile environment*/
|
||||||
#ifdef __NVCC__
|
#ifdef __NVCC__
|
||||||
#pragma push
|
#pragma push
|
||||||
#ifdef __NVCC_DIAG_PRAGMA_SUPPORT__
|
#if (__CUDACC_VER_MAJOR__ >= 11) && (__CUDACC_VER_MINOR__ >= 5)
|
||||||
#pragma nv_diag_suppress code_is_unreachable
|
#pragma nv_diag_suppress code_is_unreachable
|
||||||
#else
|
#else
|
||||||
#pragma diag_suppress code_is_unreachable
|
#pragma diag_suppress code_is_unreachable
|
||||||
|
@ -54,7 +54,6 @@ NAMESPACE_CHECK(BiCGSTAB);
|
|||||||
#include <Grid/algorithms/iterative/SchurRedBlack.h>
|
#include <Grid/algorithms/iterative/SchurRedBlack.h>
|
||||||
#include <Grid/algorithms/iterative/ConjugateGradientMultiShift.h>
|
#include <Grid/algorithms/iterative/ConjugateGradientMultiShift.h>
|
||||||
#include <Grid/algorithms/iterative/ConjugateGradientMixedPrec.h>
|
#include <Grid/algorithms/iterative/ConjugateGradientMixedPrec.h>
|
||||||
#include <Grid/algorithms/iterative/ConjugateGradientMixedPrecBatched.h>
|
|
||||||
#include <Grid/algorithms/iterative/BiCGSTABMixedPrec.h>
|
#include <Grid/algorithms/iterative/BiCGSTABMixedPrec.h>
|
||||||
#include <Grid/algorithms/iterative/BlockConjugateGradient.h>
|
#include <Grid/algorithms/iterative/BlockConjugateGradient.h>
|
||||||
#include <Grid/algorithms/iterative/ConjugateGradientReliableUpdate.h>
|
#include <Grid/algorithms/iterative/ConjugateGradientReliableUpdate.h>
|
||||||
|
@ -1,213 +0,0 @@
|
|||||||
/*************************************************************************************
|
|
||||||
|
|
||||||
Grid physics library, www.github.com/paboyle/Grid
|
|
||||||
|
|
||||||
Source file: ./lib/algorithms/iterative/ConjugateGradientMixedPrecBatched.h
|
|
||||||
|
|
||||||
Copyright (C) 2015
|
|
||||||
|
|
||||||
Author: Raoul Hodgson <raoul.hodgson@ed.ac.uk>
|
|
||||||
|
|
||||||
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_CONJUGATE_GRADIENT_MIXED_PREC_BATCHED_H
|
|
||||||
#define GRID_CONJUGATE_GRADIENT_MIXED_PREC_BATCHED_H
|
|
||||||
|
|
||||||
NAMESPACE_BEGIN(Grid);
|
|
||||||
|
|
||||||
//Mixed precision restarted defect correction CG
|
|
||||||
template<class FieldD,class FieldF,
|
|
||||||
typename std::enable_if< getPrecision<FieldD>::value == 2, int>::type = 0,
|
|
||||||
typename std::enable_if< getPrecision<FieldF>::value == 1, int>::type = 0>
|
|
||||||
class MixedPrecisionConjugateGradientBatched : public LinearFunction<FieldD> {
|
|
||||||
public:
|
|
||||||
using LinearFunction<FieldD>::operator();
|
|
||||||
RealD Tolerance;
|
|
||||||
RealD InnerTolerance; //Initial tolerance for inner CG. Defaults to Tolerance but can be changed
|
|
||||||
Integer MaxInnerIterations;
|
|
||||||
Integer MaxOuterIterations;
|
|
||||||
Integer MaxPatchupIterations;
|
|
||||||
GridBase* SinglePrecGrid; //Grid for single-precision fields
|
|
||||||
RealD OuterLoopNormMult; //Stop the outer loop and move to a final double prec solve when the residual is OuterLoopNormMult * Tolerance
|
|
||||||
LinearOperatorBase<FieldF> &Linop_f;
|
|
||||||
LinearOperatorBase<FieldD> &Linop_d;
|
|
||||||
|
|
||||||
//Option to speed up *inner single precision* solves using a LinearFunction that produces a guess
|
|
||||||
LinearFunction<FieldF> *guesser;
|
|
||||||
bool updateResidual;
|
|
||||||
|
|
||||||
MixedPrecisionConjugateGradientBatched(RealD tol,
|
|
||||||
Integer maxinnerit,
|
|
||||||
Integer maxouterit,
|
|
||||||
Integer maxpatchit,
|
|
||||||
GridBase* _sp_grid,
|
|
||||||
LinearOperatorBase<FieldF> &_Linop_f,
|
|
||||||
LinearOperatorBase<FieldD> &_Linop_d,
|
|
||||||
bool _updateResidual=true) :
|
|
||||||
Linop_f(_Linop_f), Linop_d(_Linop_d),
|
|
||||||
Tolerance(tol), InnerTolerance(tol), MaxInnerIterations(maxinnerit), MaxOuterIterations(maxouterit), MaxPatchupIterations(maxpatchit), SinglePrecGrid(_sp_grid),
|
|
||||||
OuterLoopNormMult(100.), guesser(NULL), updateResidual(_updateResidual) { };
|
|
||||||
|
|
||||||
void useGuesser(LinearFunction<FieldF> &g){
|
|
||||||
guesser = &g;
|
|
||||||
}
|
|
||||||
|
|
||||||
void operator() (const FieldD &src_d_in, FieldD &sol_d){
|
|
||||||
std::vector<FieldD> srcs_d_in{src_d_in};
|
|
||||||
std::vector<FieldD> sols_d{sol_d};
|
|
||||||
|
|
||||||
(*this)(srcs_d_in,sols_d);
|
|
||||||
|
|
||||||
sol_d = sols_d[0];
|
|
||||||
}
|
|
||||||
|
|
||||||
void operator() (const std::vector<FieldD> &src_d_in, std::vector<FieldD> &sol_d){
|
|
||||||
assert(src_d_in.size() == sol_d.size());
|
|
||||||
int NBatch = src_d_in.size();
|
|
||||||
|
|
||||||
std::cout << GridLogMessage << "NBatch = " << NBatch << std::endl;
|
|
||||||
|
|
||||||
Integer TotalOuterIterations = 0; //Number of restarts
|
|
||||||
std::vector<Integer> TotalInnerIterations(NBatch,0); //Number of inner CG iterations
|
|
||||||
std::vector<Integer> TotalFinalStepIterations(NBatch,0); //Number of CG iterations in final patch-up step
|
|
||||||
|
|
||||||
GridStopWatch TotalTimer;
|
|
||||||
TotalTimer.Start();
|
|
||||||
|
|
||||||
GridStopWatch InnerCGtimer;
|
|
||||||
GridStopWatch PrecChangeTimer;
|
|
||||||
|
|
||||||
int cb = src_d_in[0].Checkerboard();
|
|
||||||
|
|
||||||
std::vector<RealD> src_norm;
|
|
||||||
std::vector<RealD> norm;
|
|
||||||
std::vector<RealD> stop;
|
|
||||||
|
|
||||||
GridBase* DoublePrecGrid = src_d_in[0].Grid();
|
|
||||||
FieldD tmp_d(DoublePrecGrid);
|
|
||||||
tmp_d.Checkerboard() = cb;
|
|
||||||
|
|
||||||
FieldD tmp2_d(DoublePrecGrid);
|
|
||||||
tmp2_d.Checkerboard() = cb;
|
|
||||||
|
|
||||||
std::vector<FieldD> src_d;
|
|
||||||
std::vector<FieldF> src_f;
|
|
||||||
std::vector<FieldF> sol_f;
|
|
||||||
|
|
||||||
for (int i=0; i<NBatch; i++) {
|
|
||||||
sol_d[i].Checkerboard() = cb;
|
|
||||||
|
|
||||||
src_norm.push_back(norm2(src_d_in[i]));
|
|
||||||
norm.push_back(0.);
|
|
||||||
stop.push_back(src_norm[i] * Tolerance*Tolerance);
|
|
||||||
|
|
||||||
src_d.push_back(src_d_in[i]); //source for next inner iteration, computed from residual during operation
|
|
||||||
|
|
||||||
src_f.push_back(SinglePrecGrid);
|
|
||||||
src_f[i].Checkerboard() = cb;
|
|
||||||
|
|
||||||
sol_f.push_back(SinglePrecGrid);
|
|
||||||
sol_f[i].Checkerboard() = cb;
|
|
||||||
}
|
|
||||||
|
|
||||||
RealD inner_tol = InnerTolerance;
|
|
||||||
|
|
||||||
ConjugateGradient<FieldF> CG_f(inner_tol, MaxInnerIterations);
|
|
||||||
CG_f.ErrorOnNoConverge = false;
|
|
||||||
|
|
||||||
Integer &outer_iter = TotalOuterIterations; //so it will be equal to the final iteration count
|
|
||||||
|
|
||||||
for(outer_iter = 0; outer_iter < MaxOuterIterations; outer_iter++){
|
|
||||||
std::cout << GridLogMessage << std::endl;
|
|
||||||
std::cout << GridLogMessage << "Outer iteration " << outer_iter << std::endl;
|
|
||||||
|
|
||||||
bool allConverged = true;
|
|
||||||
|
|
||||||
for (int i=0; i<NBatch; i++) {
|
|
||||||
//Compute double precision rsd and also new RHS vector.
|
|
||||||
Linop_d.HermOp(sol_d[i], tmp_d);
|
|
||||||
norm[i] = axpy_norm(src_d[i], -1., tmp_d, src_d_in[i]); //src_d is residual vector
|
|
||||||
|
|
||||||
std::cout<<GridLogMessage<<"MixedPrecisionConjugateGradientBatched: Outer iteration " << outer_iter <<" solve " << i << " residual "<< norm[i] << " target "<< stop[i] <<std::endl;
|
|
||||||
|
|
||||||
PrecChangeTimer.Start();
|
|
||||||
precisionChange(src_f[i], src_d[i]);
|
|
||||||
PrecChangeTimer.Stop();
|
|
||||||
|
|
||||||
sol_f[i] = Zero();
|
|
||||||
|
|
||||||
if(norm[i] > OuterLoopNormMult * stop[i]) {
|
|
||||||
allConverged = false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (allConverged) break;
|
|
||||||
|
|
||||||
if (updateResidual) {
|
|
||||||
RealD normMax = *std::max_element(std::begin(norm), std::end(norm));
|
|
||||||
RealD stopMax = *std::max_element(std::begin(stop), std::end(stop));
|
|
||||||
while( normMax * inner_tol * inner_tol < stopMax) inner_tol *= 2; // inner_tol = sqrt(stop/norm) ??
|
|
||||||
CG_f.Tolerance = inner_tol;
|
|
||||||
}
|
|
||||||
|
|
||||||
//Optionally improve inner solver guess (eg using known eigenvectors)
|
|
||||||
if(guesser != NULL) {
|
|
||||||
(*guesser)(src_f, sol_f);
|
|
||||||
}
|
|
||||||
|
|
||||||
for (int i=0; i<NBatch; i++) {
|
|
||||||
//Inner CG
|
|
||||||
InnerCGtimer.Start();
|
|
||||||
CG_f(Linop_f, src_f[i], sol_f[i]);
|
|
||||||
InnerCGtimer.Stop();
|
|
||||||
TotalInnerIterations[i] += CG_f.IterationsToComplete;
|
|
||||||
|
|
||||||
//Convert sol back to double and add to double prec solution
|
|
||||||
PrecChangeTimer.Start();
|
|
||||||
precisionChange(tmp_d, sol_f[i]);
|
|
||||||
PrecChangeTimer.Stop();
|
|
||||||
|
|
||||||
axpy(sol_d[i], 1.0, tmp_d, sol_d[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
//Final trial CG
|
|
||||||
std::cout << GridLogMessage << std::endl;
|
|
||||||
std::cout<<GridLogMessage<<"MixedPrecisionConjugateGradientBatched: Starting final patch-up double-precision solve"<<std::endl;
|
|
||||||
|
|
||||||
for (int i=0; i<NBatch; i++) {
|
|
||||||
ConjugateGradient<FieldD> CG_d(Tolerance, MaxPatchupIterations);
|
|
||||||
CG_d(Linop_d, src_d_in[i], sol_d[i]);
|
|
||||||
TotalFinalStepIterations[i] += CG_d.IterationsToComplete;
|
|
||||||
}
|
|
||||||
|
|
||||||
TotalTimer.Stop();
|
|
||||||
|
|
||||||
std::cout << GridLogMessage << std::endl;
|
|
||||||
for (int i=0; i<NBatch; i++) {
|
|
||||||
std::cout<<GridLogMessage<<"MixedPrecisionConjugateGradientBatched: solve " << i << " Inner CG iterations " << TotalInnerIterations[i] << " Restarts " << TotalOuterIterations << " Final CG iterations " << TotalFinalStepIterations[i] << std::endl;
|
|
||||||
}
|
|
||||||
std::cout << GridLogMessage << std::endl;
|
|
||||||
std::cout<<GridLogMessage<<"MixedPrecisionConjugateGradientBatched: Total time " << TotalTimer.Elapsed() << " Precision change " << PrecChangeTimer.Elapsed() << " Inner CG total " << InnerCGtimer.Elapsed() << std::endl;
|
|
||||||
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
NAMESPACE_END(Grid);
|
|
||||||
|
|
||||||
#endif
|
|
@ -4,14 +4,11 @@ NAMESPACE_BEGIN(Grid);
|
|||||||
|
|
||||||
/*Allocation types, saying which pointer cache should be used*/
|
/*Allocation types, saying which pointer cache should be used*/
|
||||||
#define Cpu (0)
|
#define Cpu (0)
|
||||||
#define CpuHuge (1)
|
#define CpuSmall (1)
|
||||||
#define CpuSmall (2)
|
#define Acc (2)
|
||||||
#define Acc (3)
|
#define AccSmall (3)
|
||||||
#define AccHuge (4)
|
#define Shared (4)
|
||||||
#define AccSmall (5)
|
#define SharedSmall (5)
|
||||||
#define Shared (6)
|
|
||||||
#define SharedHuge (7)
|
|
||||||
#define SharedSmall (8)
|
|
||||||
#undef GRID_MM_VERBOSE
|
#undef GRID_MM_VERBOSE
|
||||||
uint64_t total_shared;
|
uint64_t total_shared;
|
||||||
uint64_t total_device;
|
uint64_t total_device;
|
||||||
@ -38,14 +35,12 @@ void MemoryManager::PrintBytes(void)
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
uint64_t MemoryManager::DeviceCacheBytes() { return CacheBytes[Acc] + CacheBytes[AccHuge] + CacheBytes[AccSmall]; }
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////
|
//////////////////////////////////////////////////////////////////////
|
||||||
// Data tables for recently freed pooiniter caches
|
// Data tables for recently freed pooiniter caches
|
||||||
//////////////////////////////////////////////////////////////////////
|
//////////////////////////////////////////////////////////////////////
|
||||||
MemoryManager::AllocationCacheEntry MemoryManager::Entries[MemoryManager::NallocType][MemoryManager::NallocCacheMax];
|
MemoryManager::AllocationCacheEntry MemoryManager::Entries[MemoryManager::NallocType][MemoryManager::NallocCacheMax];
|
||||||
int MemoryManager::Victim[MemoryManager::NallocType];
|
int MemoryManager::Victim[MemoryManager::NallocType];
|
||||||
int MemoryManager::Ncache[MemoryManager::NallocType] = { 2, 0, 8, 8, 0, 16, 8, 0, 16 };
|
int MemoryManager::Ncache[MemoryManager::NallocType] = { 2, 8, 8, 16, 8, 16 };
|
||||||
uint64_t MemoryManager::CacheBytes[MemoryManager::NallocType];
|
uint64_t MemoryManager::CacheBytes[MemoryManager::NallocType];
|
||||||
//////////////////////////////////////////////////////////////////////
|
//////////////////////////////////////////////////////////////////////
|
||||||
// Actual allocation and deallocation utils
|
// Actual allocation and deallocation utils
|
||||||
@ -175,16 +170,6 @@ void MemoryManager::Init(void)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
str= getenv("GRID_ALLOC_NCACHE_HUGE");
|
|
||||||
if ( str ) {
|
|
||||||
Nc = atoi(str);
|
|
||||||
if ( (Nc>=0) && (Nc < NallocCacheMax)) {
|
|
||||||
Ncache[CpuHuge]=Nc;
|
|
||||||
Ncache[AccHuge]=Nc;
|
|
||||||
Ncache[SharedHuge]=Nc;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
str= getenv("GRID_ALLOC_NCACHE_SMALL");
|
str= getenv("GRID_ALLOC_NCACHE_SMALL");
|
||||||
if ( str ) {
|
if ( str ) {
|
||||||
Nc = atoi(str);
|
Nc = atoi(str);
|
||||||
@ -205,9 +190,7 @@ void MemoryManager::InitMessage(void) {
|
|||||||
|
|
||||||
std::cout << GridLogMessage<< "MemoryManager::Init() setting up"<<std::endl;
|
std::cout << GridLogMessage<< "MemoryManager::Init() setting up"<<std::endl;
|
||||||
#ifdef ALLOCATION_CACHE
|
#ifdef ALLOCATION_CACHE
|
||||||
std::cout << GridLogMessage<< "MemoryManager::Init() cache pool for recent host allocations: SMALL "<<Ncache[CpuSmall]<<" LARGE "<<Ncache[Cpu]<<" HUGE "<<Ncache[CpuHuge]<<std::endl;
|
std::cout << GridLogMessage<< "MemoryManager::Init() cache pool for recent allocations: SMALL "<<Ncache[CpuSmall]<<" LARGE "<<Ncache[Cpu]<<std::endl;
|
||||||
std::cout << GridLogMessage<< "MemoryManager::Init() cache pool for recent device allocations: SMALL "<<Ncache[AccSmall]<<" LARGE "<<Ncache[Acc]<<" Huge "<<Ncache[AccHuge]<<std::endl;
|
|
||||||
std::cout << GridLogMessage<< "MemoryManager::Init() cache pool for recent shared allocations: SMALL "<<Ncache[SharedSmall]<<" LARGE "<<Ncache[Shared]<<" Huge "<<Ncache[SharedHuge]<<std::endl;
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef GRID_UVM
|
#ifdef GRID_UVM
|
||||||
@ -239,11 +222,8 @@ void MemoryManager::InitMessage(void) {
|
|||||||
void *MemoryManager::Insert(void *ptr,size_t bytes,int type)
|
void *MemoryManager::Insert(void *ptr,size_t bytes,int type)
|
||||||
{
|
{
|
||||||
#ifdef ALLOCATION_CACHE
|
#ifdef ALLOCATION_CACHE
|
||||||
int cache;
|
bool small = (bytes < GRID_ALLOC_SMALL_LIMIT);
|
||||||
if (bytes < GRID_ALLOC_SMALL_LIMIT) cache = type + 2;
|
int cache = type + small;
|
||||||
else if (bytes >= GRID_ALLOC_HUGE_LIMIT) cache = type + 1;
|
|
||||||
else cache = type;
|
|
||||||
|
|
||||||
return Insert(ptr,bytes,Entries[cache],Ncache[cache],Victim[cache],CacheBytes[cache]);
|
return Insert(ptr,bytes,Entries[cache],Ncache[cache],Victim[cache],CacheBytes[cache]);
|
||||||
#else
|
#else
|
||||||
return ptr;
|
return ptr;
|
||||||
@ -252,12 +232,11 @@ void *MemoryManager::Insert(void *ptr,size_t bytes,int type)
|
|||||||
|
|
||||||
void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim, uint64_t &cacheBytes)
|
void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries,int ncache,int &victim, uint64_t &cacheBytes)
|
||||||
{
|
{
|
||||||
|
assert(ncache>0);
|
||||||
#ifdef GRID_OMP
|
#ifdef GRID_OMP
|
||||||
assert(omp_in_parallel()==0);
|
assert(omp_in_parallel()==0);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (ncache == 0) return ptr;
|
|
||||||
|
|
||||||
void * ret = NULL;
|
void * ret = NULL;
|
||||||
int v = -1;
|
int v = -1;
|
||||||
|
|
||||||
@ -292,11 +271,8 @@ void *MemoryManager::Insert(void *ptr,size_t bytes,AllocationCacheEntry *entries
|
|||||||
void *MemoryManager::Lookup(size_t bytes,int type)
|
void *MemoryManager::Lookup(size_t bytes,int type)
|
||||||
{
|
{
|
||||||
#ifdef ALLOCATION_CACHE
|
#ifdef ALLOCATION_CACHE
|
||||||
int cache;
|
bool small = (bytes < GRID_ALLOC_SMALL_LIMIT);
|
||||||
if (bytes < GRID_ALLOC_SMALL_LIMIT) cache = type + 2;
|
int cache = type+small;
|
||||||
else if (bytes >= GRID_ALLOC_HUGE_LIMIT) cache = type + 1;
|
|
||||||
else cache = type;
|
|
||||||
|
|
||||||
return Lookup(bytes,Entries[cache],Ncache[cache],CacheBytes[cache]);
|
return Lookup(bytes,Entries[cache],Ncache[cache],CacheBytes[cache]);
|
||||||
#else
|
#else
|
||||||
return NULL;
|
return NULL;
|
||||||
@ -305,6 +281,7 @@ void *MemoryManager::Lookup(size_t bytes,int type)
|
|||||||
|
|
||||||
void *MemoryManager::Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache,uint64_t & cacheBytes)
|
void *MemoryManager::Lookup(size_t bytes,AllocationCacheEntry *entries,int ncache,uint64_t & cacheBytes)
|
||||||
{
|
{
|
||||||
|
assert(ncache>0);
|
||||||
#ifdef GRID_OMP
|
#ifdef GRID_OMP
|
||||||
assert(omp_in_parallel()==0);
|
assert(omp_in_parallel()==0);
|
||||||
#endif
|
#endif
|
||||||
|
@ -35,7 +35,6 @@ NAMESPACE_BEGIN(Grid);
|
|||||||
// Move control to configure.ac and Config.h?
|
// Move control to configure.ac and Config.h?
|
||||||
|
|
||||||
#define GRID_ALLOC_SMALL_LIMIT (4096)
|
#define GRID_ALLOC_SMALL_LIMIT (4096)
|
||||||
#define GRID_ALLOC_HUGE_LIMIT (2147483648)
|
|
||||||
|
|
||||||
#define STRINGIFY(x) #x
|
#define STRINGIFY(x) #x
|
||||||
#define TOSTRING(x) STRINGIFY(x)
|
#define TOSTRING(x) STRINGIFY(x)
|
||||||
@ -84,7 +83,7 @@ private:
|
|||||||
} AllocationCacheEntry;
|
} AllocationCacheEntry;
|
||||||
|
|
||||||
static const int NallocCacheMax=128;
|
static const int NallocCacheMax=128;
|
||||||
static const int NallocType=9;
|
static const int NallocType=6;
|
||||||
static AllocationCacheEntry Entries[NallocType][NallocCacheMax];
|
static AllocationCacheEntry Entries[NallocType][NallocCacheMax];
|
||||||
static int Victim[NallocType];
|
static int Victim[NallocType];
|
||||||
static int Ncache[NallocType];
|
static int Ncache[NallocType];
|
||||||
@ -122,9 +121,7 @@ private:
|
|||||||
static uint64_t DeviceToHostXfer;
|
static uint64_t DeviceToHostXfer;
|
||||||
static uint64_t DeviceEvictions;
|
static uint64_t DeviceEvictions;
|
||||||
static uint64_t DeviceDestroy;
|
static uint64_t DeviceDestroy;
|
||||||
|
|
||||||
static uint64_t DeviceCacheBytes();
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
#ifndef GRID_UVM
|
#ifndef GRID_UVM
|
||||||
//////////////////////////////////////////////////////////////////////
|
//////////////////////////////////////////////////////////////////////
|
||||||
|
@ -46,4 +46,3 @@ Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|||||||
#include <Grid/lattice/Lattice_unary.h>
|
#include <Grid/lattice/Lattice_unary.h>
|
||||||
#include <Grid/lattice/Lattice_transfer.h>
|
#include <Grid/lattice/Lattice_transfer.h>
|
||||||
#include <Grid/lattice/Lattice_basis.h>
|
#include <Grid/lattice/Lattice_basis.h>
|
||||||
#include <Grid/lattice/Lattice_crc.h>
|
|
||||||
|
@ -129,7 +129,7 @@ public:
|
|||||||
|
|
||||||
auto exprCopy = expr;
|
auto exprCopy = expr;
|
||||||
ExpressionViewOpen(exprCopy);
|
ExpressionViewOpen(exprCopy);
|
||||||
auto me = View(AcceleratorWriteDiscard);
|
auto me = View(AcceleratorWrite);
|
||||||
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
||||||
auto tmp = eval(ss,exprCopy);
|
auto tmp = eval(ss,exprCopy);
|
||||||
coalescedWrite(me[ss],tmp);
|
coalescedWrite(me[ss],tmp);
|
||||||
@ -152,7 +152,7 @@ public:
|
|||||||
|
|
||||||
auto exprCopy = expr;
|
auto exprCopy = expr;
|
||||||
ExpressionViewOpen(exprCopy);
|
ExpressionViewOpen(exprCopy);
|
||||||
auto me = View(AcceleratorWriteDiscard);
|
auto me = View(AcceleratorWrite);
|
||||||
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
||||||
auto tmp = eval(ss,exprCopy);
|
auto tmp = eval(ss,exprCopy);
|
||||||
coalescedWrite(me[ss],tmp);
|
coalescedWrite(me[ss],tmp);
|
||||||
@ -174,7 +174,7 @@ public:
|
|||||||
this->checkerboard=cb;
|
this->checkerboard=cb;
|
||||||
auto exprCopy = expr;
|
auto exprCopy = expr;
|
||||||
ExpressionViewOpen(exprCopy);
|
ExpressionViewOpen(exprCopy);
|
||||||
auto me = View(AcceleratorWriteDiscard);
|
auto me = View(AcceleratorWrite);
|
||||||
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
||||||
auto tmp = eval(ss,exprCopy);
|
auto tmp = eval(ss,exprCopy);
|
||||||
coalescedWrite(me[ss],tmp);
|
coalescedWrite(me[ss],tmp);
|
||||||
@ -288,8 +288,8 @@ public:
|
|||||||
typename std::enable_if<!std::is_same<robj,vobj>::value,int>::type i=0;
|
typename std::enable_if<!std::is_same<robj,vobj>::value,int>::type i=0;
|
||||||
conformable(*this,r);
|
conformable(*this,r);
|
||||||
this->checkerboard = r.Checkerboard();
|
this->checkerboard = r.Checkerboard();
|
||||||
|
auto me = View(AcceleratorWrite);
|
||||||
auto him= r.View(AcceleratorRead);
|
auto him= r.View(AcceleratorRead);
|
||||||
auto me = View(AcceleratorWriteDiscard);
|
|
||||||
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
||||||
coalescedWrite(me[ss],him(ss));
|
coalescedWrite(me[ss],him(ss));
|
||||||
});
|
});
|
||||||
@ -303,8 +303,8 @@ public:
|
|||||||
inline Lattice<vobj> & operator = (const Lattice<vobj> & r){
|
inline Lattice<vobj> & operator = (const Lattice<vobj> & r){
|
||||||
this->checkerboard = r.Checkerboard();
|
this->checkerboard = r.Checkerboard();
|
||||||
conformable(*this,r);
|
conformable(*this,r);
|
||||||
|
auto me = View(AcceleratorWrite);
|
||||||
auto him= r.View(AcceleratorRead);
|
auto him= r.View(AcceleratorRead);
|
||||||
auto me = View(AcceleratorWriteDiscard);
|
|
||||||
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
accelerator_for(ss,me.size(),vobj::Nsimd(),{
|
||||||
coalescedWrite(me[ss],him(ss));
|
coalescedWrite(me[ss],him(ss));
|
||||||
});
|
});
|
||||||
|
@ -1,55 +0,0 @@
|
|||||||
/*************************************************************************************
|
|
||||||
|
|
||||||
Grid physics library, www.github.com/paboyle/Grid
|
|
||||||
|
|
||||||
Source file: ./lib/lattice/Lattice_crc.h
|
|
||||||
|
|
||||||
Copyright (C) 2021
|
|
||||||
|
|
||||||
Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|
||||||
|
|
||||||
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 */
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
NAMESPACE_BEGIN(Grid);
|
|
||||||
|
|
||||||
template<class vobj> void DumpSliceNorm(std::string s,Lattice<vobj> &f,int mu=-1)
|
|
||||||
{
|
|
||||||
auto ff = localNorm2(f);
|
|
||||||
if ( mu==-1 ) mu = f.Grid()->Nd()-1;
|
|
||||||
typedef typename vobj::tensor_reduced normtype;
|
|
||||||
typedef typename normtype::scalar_object scalar;
|
|
||||||
std::vector<scalar> sff;
|
|
||||||
sliceSum(ff,sff,mu);
|
|
||||||
for(int t=0;t<sff.size();t++){
|
|
||||||
std::cout << s<<" "<<t<<" "<<sff[t]<<std::endl;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<class vobj> uint32_t crc(Lattice<vobj> & buf)
|
|
||||||
{
|
|
||||||
autoView( buf_v , buf, CpuRead);
|
|
||||||
return ::crc32(0L,(unsigned char *)&buf_v[0],(size_t)sizeof(vobj)*buf.oSites());
|
|
||||||
}
|
|
||||||
|
|
||||||
#define CRC(U) std::cout << "FingerPrint "<<__FILE__ <<" "<< __LINE__ <<" "<< #U <<" "<<crc(U)<<std::endl;
|
|
||||||
|
|
||||||
NAMESPACE_END(Grid);
|
|
||||||
|
|
||||||
|
|
@ -288,36 +288,7 @@ inline void blockProject(Lattice<iVector<CComplex,nbasis > > &coarseData,
|
|||||||
blockZAXPY(fineDataRed,ip,Basis[v],fineDataRed);
|
blockZAXPY(fineDataRed,ip,Basis[v],fineDataRed);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
template<class vobj,class CComplex,int nbasis,class VLattice>
|
|
||||||
inline void batchBlockProject(std::vector<Lattice<iVector<CComplex,nbasis>>> &coarseData,
|
|
||||||
const std::vector<Lattice<vobj>> &fineData,
|
|
||||||
const VLattice &Basis)
|
|
||||||
{
|
|
||||||
int NBatch = fineData.size();
|
|
||||||
assert(coarseData.size() == NBatch);
|
|
||||||
|
|
||||||
GridBase * fine = fineData[0].Grid();
|
|
||||||
GridBase * coarse= coarseData[0].Grid();
|
|
||||||
|
|
||||||
Lattice<iScalar<CComplex>> ip(coarse);
|
|
||||||
std::vector<Lattice<vobj>> fineDataCopy = fineData;
|
|
||||||
|
|
||||||
autoView(ip_, ip, AcceleratorWrite);
|
|
||||||
for(int v=0;v<nbasis;v++) {
|
|
||||||
for (int k=0; k<NBatch; k++) {
|
|
||||||
autoView( coarseData_ , coarseData[k], AcceleratorWrite);
|
|
||||||
blockInnerProductD(ip,Basis[v],fineDataCopy[k]); // ip = <basis|fine>
|
|
||||||
accelerator_for( sc, coarse->oSites(), vobj::Nsimd(), {
|
|
||||||
convertType(coarseData_[sc](v),ip_[sc]);
|
|
||||||
});
|
|
||||||
|
|
||||||
// improve numerical stability of projection
|
|
||||||
// |fine> = |fine> - <basis|fine> |basis>
|
|
||||||
ip=-ip;
|
|
||||||
blockZAXPY(fineDataCopy[k],ip,Basis[v],fineDataCopy[k]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<class vobj,class vobj2,class CComplex>
|
template<class vobj,class vobj2,class CComplex>
|
||||||
inline void blockZAXPY(Lattice<vobj> &fineZ,
|
inline void blockZAXPY(Lattice<vobj> &fineZ,
|
||||||
@ -619,26 +590,6 @@ inline void blockPromote(const Lattice<iVector<CComplex,nbasis > > &coarseData,
|
|||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
template<class vobj,class CComplex,int nbasis,class VLattice>
|
|
||||||
inline void batchBlockPromote(const std::vector<Lattice<iVector<CComplex,nbasis>>> &coarseData,
|
|
||||||
std::vector<Lattice<vobj>> &fineData,
|
|
||||||
const VLattice &Basis)
|
|
||||||
{
|
|
||||||
int NBatch = coarseData.size();
|
|
||||||
assert(fineData.size() == NBatch);
|
|
||||||
|
|
||||||
GridBase * fine = fineData[0].Grid();
|
|
||||||
GridBase * coarse = coarseData[0].Grid();
|
|
||||||
for (int k=0; k<NBatch; k++)
|
|
||||||
fineData[k]=Zero();
|
|
||||||
for (int i=0;i<nbasis;i++) {
|
|
||||||
for (int k=0; k<NBatch; k++) {
|
|
||||||
Lattice<iScalar<CComplex>> ip = PeekIndex<0>(coarseData[k],i);
|
|
||||||
blockZAXPY(fineData[k],ip,Basis[i],fineData[k]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Useful for precision conversion, or indeed anything where an operator= does a conversion on scalars.
|
// Useful for precision conversion, or indeed anything where an operator= does a conversion on scalars.
|
||||||
// Simd layouts need not match since we use peek/poke Local
|
// Simd layouts need not match since we use peek/poke Local
|
||||||
template<class vobj,class vvobj>
|
template<class vobj,class vvobj>
|
||||||
|
@ -16,7 +16,7 @@
|
|||||||
|
|
||||||
#ifdef __NVCC__
|
#ifdef __NVCC__
|
||||||
#pragma push
|
#pragma push
|
||||||
#ifdef __NVCC_DIAG_PRAGMA_SUPPORT__
|
#if (__CUDACC_VER_MAJOR__ >= 11) && (__CUDACC_VER_MINOR__ >= 5)
|
||||||
#pragma nv_diag_suppress declared_but_not_referenced // suppress "function was declared but never referenced warning"
|
#pragma nv_diag_suppress declared_but_not_referenced // suppress "function was declared but never referenced warning"
|
||||||
#else
|
#else
|
||||||
#pragma diag_suppress declared_but_not_referenced // suppress "function was declared but never referenced warning"
|
#pragma diag_suppress declared_but_not_referenced // suppress "function was declared but never referenced warning"
|
||||||
|
@ -451,20 +451,9 @@ template<class vobj> void pokeLorentz(vobj &lhs,const decltype(peekIndex<Lorentz
|
|||||||
// Fermion <-> propagator assignements
|
// Fermion <-> propagator assignements
|
||||||
//////////////////////////////////////////////
|
//////////////////////////////////////////////
|
||||||
//template <class Prop, class Ferm>
|
//template <class Prop, class Ferm>
|
||||||
#define FAST_FERM_TO_PROP
|
|
||||||
template <class Fimpl>
|
template <class Fimpl>
|
||||||
void FermToProp(typename Fimpl::PropagatorField &p, const typename Fimpl::FermionField &f, const int s, const int c)
|
void FermToProp(typename Fimpl::PropagatorField &p, const typename Fimpl::FermionField &f, const int s, const int c)
|
||||||
{
|
{
|
||||||
#ifdef FAST_FERM_TO_PROP
|
|
||||||
autoView(p_v,p,CpuWrite);
|
|
||||||
autoView(f_v,f,CpuRead);
|
|
||||||
thread_for(idx,p_v.oSites(),{
|
|
||||||
for(int ss = 0; ss < Ns; ++ss) {
|
|
||||||
for(int cc = 0; cc < Fimpl::Dimension; ++cc) {
|
|
||||||
p_v[idx]()(ss,s)(cc,c) = f_v[idx]()(ss)(cc); // Propagator sink index is LEFT, suitable for left mult by gauge link (e.g.)
|
|
||||||
}}
|
|
||||||
});
|
|
||||||
#else
|
|
||||||
for(int j = 0; j < Ns; ++j)
|
for(int j = 0; j < Ns; ++j)
|
||||||
{
|
{
|
||||||
auto pjs = peekSpin(p, j, s);
|
auto pjs = peekSpin(p, j, s);
|
||||||
@ -476,23 +465,12 @@ void FermToProp(typename Fimpl::PropagatorField &p, const typename Fimpl::Fermio
|
|||||||
}
|
}
|
||||||
pokeSpin(p, pjs, j, s);
|
pokeSpin(p, pjs, j, s);
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//template <class Prop, class Ferm>
|
//template <class Prop, class Ferm>
|
||||||
template <class Fimpl>
|
template <class Fimpl>
|
||||||
void PropToFerm(typename Fimpl::FermionField &f, const typename Fimpl::PropagatorField &p, const int s, const int c)
|
void PropToFerm(typename Fimpl::FermionField &f, const typename Fimpl::PropagatorField &p, const int s, const int c)
|
||||||
{
|
{
|
||||||
#ifdef FAST_FERM_TO_PROP
|
|
||||||
autoView(p_v,p,CpuRead);
|
|
||||||
autoView(f_v,f,CpuWrite);
|
|
||||||
thread_for(idx,p_v.oSites(),{
|
|
||||||
for(int ss = 0; ss < Ns; ++ss) {
|
|
||||||
for(int cc = 0; cc < Fimpl::Dimension; ++cc) {
|
|
||||||
f_v[idx]()(ss)(cc) = p_v[idx]()(ss,s)(cc,c); // LEFT index is copied across for s,c right index
|
|
||||||
}}
|
|
||||||
});
|
|
||||||
#else
|
|
||||||
for(int j = 0; j < Ns; ++j)
|
for(int j = 0; j < Ns; ++j)
|
||||||
{
|
{
|
||||||
auto pjs = peekSpin(p, j, s);
|
auto pjs = peekSpin(p, j, s);
|
||||||
@ -504,7 +482,6 @@ void PropToFerm(typename Fimpl::FermionField &f, const typename Fimpl::Propagato
|
|||||||
}
|
}
|
||||||
pokeSpin(f, fj, j);
|
pokeSpin(f, fj, j);
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
//////////////////////////////////////////////
|
//////////////////////////////////////////////
|
||||||
|
@ -1,270 +0,0 @@
|
|||||||
/*************************************************************************************
|
|
||||||
grid` physics library, www.github.com/paboyle/Grid
|
|
||||||
|
|
||||||
Source file: ./tests/Test_cshift.cc
|
|
||||||
|
|
||||||
Copyright (C) 2015
|
|
||||||
|
|
||||||
Author: Azusa Yamaguchi <ayamaguc@staffmail.ed.ac.uk>
|
|
||||||
Author: Peter Boyle <paboyle@ph.ed.ac.uk>
|
|
||||||
|
|
||||||
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 */
|
|
||||||
#include <Grid/Grid.h>
|
|
||||||
|
|
||||||
using namespace Grid;
|
|
||||||
|
|
||||||
Gamma::Algebra Gmu [] = {
|
|
||||||
Gamma::Algebra::GammaX,
|
|
||||||
Gamma::Algebra::GammaY,
|
|
||||||
Gamma::Algebra::GammaZ,
|
|
||||||
Gamma::Algebra::GammaT,
|
|
||||||
Gamma::Algebra::Gamma5
|
|
||||||
};
|
|
||||||
|
|
||||||
int main (int argc, char ** argv)
|
|
||||||
{
|
|
||||||
Grid_init(&argc,&argv);
|
|
||||||
|
|
||||||
int threads = GridThread::GetThreads();
|
|
||||||
std::cout<<GridLogMessage << "Grid is setup to use "<<threads<<" threads"<<std::endl;
|
|
||||||
|
|
||||||
Coordinate latt_size = GridDefaultLatt();
|
|
||||||
Coordinate simd_layout = GridDefaultSimd(Nd,vComplexD::Nsimd());
|
|
||||||
Coordinate mpi_layout = GridDefaultMpi();
|
|
||||||
|
|
||||||
int vol = 1;
|
|
||||||
for(int d=0;d<latt_size.size();d++){
|
|
||||||
vol = vol * latt_size[d];
|
|
||||||
}
|
|
||||||
GridCartesian GRID(latt_size,simd_layout,mpi_layout);
|
|
||||||
GridRedBlackCartesian RBGRID(&GRID);
|
|
||||||
|
|
||||||
LatticeComplexD coor(&GRID);
|
|
||||||
ComplexD ci(0.0,1.0);
|
|
||||||
|
|
||||||
std::vector<int> seeds({1,2,3,4});
|
|
||||||
GridSerialRNG sRNG; sRNG.SeedFixedIntegers(seeds); // naughty seeding
|
|
||||||
GridParallelRNG pRNG(&GRID);
|
|
||||||
pRNG.SeedFixedIntegers(seeds);
|
|
||||||
|
|
||||||
LatticeGaugeFieldD Umu(&GRID);
|
|
||||||
SU<Nc>::ColdConfiguration(pRNG,Umu); // Unit gauge
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
// Wilson test
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
{
|
|
||||||
LatticeFermionD src(&GRID); gaussian(pRNG,src);
|
|
||||||
LatticeFermionD src_p(&GRID);
|
|
||||||
LatticeFermionD tmp(&GRID);
|
|
||||||
LatticeFermionD ref(&GRID);
|
|
||||||
LatticeFermionD result(&GRID);
|
|
||||||
|
|
||||||
RealD mass=0.1;
|
|
||||||
WilsonFermionD Dw(Umu,GRID,RBGRID,mass);
|
|
||||||
|
|
||||||
Dw.M(src,ref);
|
|
||||||
std::cout << "Norm src "<<norm2(src)<<std::endl;
|
|
||||||
std::cout << "Norm Dw x src "<<norm2(ref)<<std::endl;
|
|
||||||
{
|
|
||||||
FFT theFFT(&GRID);
|
|
||||||
|
|
||||||
////////////////
|
|
||||||
// operator in Fourier space
|
|
||||||
////////////////
|
|
||||||
tmp =ref;
|
|
||||||
theFFT.FFT_all_dim(result,tmp,FFT::forward);
|
|
||||||
std::cout<<"FFT[ Dw x src ] "<< norm2(result)<<std::endl;
|
|
||||||
|
|
||||||
tmp = src;
|
|
||||||
theFFT.FFT_all_dim(src_p,tmp,FFT::forward);
|
|
||||||
std::cout<<"FFT[ src ] "<< norm2(src_p)<<std::endl;
|
|
||||||
|
|
||||||
/////////////////////////////////////////////////////////////////
|
|
||||||
// work out the predicted FT from Fourier
|
|
||||||
/////////////////////////////////////////////////////////////////
|
|
||||||
auto FGrid = &GRID;
|
|
||||||
LatticeFermionD Kinetic(FGrid); Kinetic = Zero();
|
|
||||||
LatticeComplexD kmu(FGrid);
|
|
||||||
LatticeInteger scoor(FGrid);
|
|
||||||
LatticeComplexD sk (FGrid); sk = Zero();
|
|
||||||
LatticeComplexD sk2(FGrid); sk2= Zero();
|
|
||||||
LatticeComplexD W(FGrid); W= Zero();
|
|
||||||
LatticeComplexD one(FGrid); one =ComplexD(1.0,0.0);
|
|
||||||
ComplexD ci(0.0,1.0);
|
|
||||||
|
|
||||||
for(int mu=0;mu<Nd;mu++) {
|
|
||||||
|
|
||||||
RealD TwoPiL = M_PI * 2.0/ latt_size[mu];
|
|
||||||
|
|
||||||
LatticeCoordinate(kmu,mu);
|
|
||||||
|
|
||||||
kmu = TwoPiL * kmu;
|
|
||||||
|
|
||||||
sk2 = sk2 + 2.0*sin(kmu*0.5)*sin(kmu*0.5);
|
|
||||||
sk = sk + sin(kmu) *sin(kmu);
|
|
||||||
|
|
||||||
// -1/2 Dw -> 1/2 gmu (eip - emip) = i sinp gmu
|
|
||||||
Kinetic = Kinetic + sin(kmu)*ci*(Gamma(Gmu[mu])*src_p);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
W = mass + sk2;
|
|
||||||
Kinetic = Kinetic + W * src_p;
|
|
||||||
|
|
||||||
std::cout<<"Momentum space src "<< norm2(src_p)<<std::endl;
|
|
||||||
std::cout<<"Momentum space Dw x src "<< norm2(Kinetic)<<std::endl;
|
|
||||||
std::cout<<"FT[Coordinate space Dw] "<< norm2(result)<<std::endl;
|
|
||||||
|
|
||||||
result = result - Kinetic;
|
|
||||||
std::cout<<"diff "<< norm2(result)<<std::endl;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << " =======================================" <<std::endl;
|
|
||||||
std::cout << " Checking FourierFreePropagator x Dw = 1" <<std::endl;
|
|
||||||
std::cout << " =======================================" <<std::endl;
|
|
||||||
std::cout << "Dw src = " <<norm2(src)<<std::endl;
|
|
||||||
std::cout << "Dw tmp = " <<norm2(tmp)<<std::endl;
|
|
||||||
Dw.M(src,tmp);
|
|
||||||
Dw.FreePropagator(tmp,ref,mass);
|
|
||||||
|
|
||||||
std::cout << "Dw ref = " <<norm2(ref)<<std::endl;
|
|
||||||
|
|
||||||
ref = ref - src;
|
|
||||||
|
|
||||||
std::cout << "Dw ref-src = " <<norm2(ref)<<std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
// Wilson prop
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
{
|
|
||||||
std::cout<<"****************************************"<<std::endl;
|
|
||||||
std::cout << "Wilson Mom space 4d propagator \n";
|
|
||||||
std::cout<<"****************************************"<<std::endl;
|
|
||||||
|
|
||||||
LatticeFermionD src(&GRID); gaussian(pRNG,src);
|
|
||||||
LatticeFermionD tmp(&GRID);
|
|
||||||
LatticeFermionD ref(&GRID);
|
|
||||||
LatticeFermionD diff(&GRID);
|
|
||||||
|
|
||||||
src=Zero();
|
|
||||||
Coordinate point(4,0); // 0,0,0,0
|
|
||||||
SpinColourVectorD ferm;
|
|
||||||
ferm=Zero();
|
|
||||||
ferm()(0)(0) = ComplexD(1.0);
|
|
||||||
pokeSite(ferm,src,point);
|
|
||||||
|
|
||||||
RealD mass=0.1;
|
|
||||||
WilsonFermionD Dw(Umu,GRID,RBGRID,mass);
|
|
||||||
|
|
||||||
// Momentum space prop
|
|
||||||
std::cout << " Solving by FFT and Feynman rules" <<std::endl;
|
|
||||||
Dw.FreePropagator(src,ref,mass) ;
|
|
||||||
|
|
||||||
Gamma G5(Gamma::Algebra::Gamma5);
|
|
||||||
|
|
||||||
LatticeFermionD result(&GRID);
|
|
||||||
const int sdir=0;
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
// Conjugate gradient on normal equations system
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
std::cout << " Solving by Conjugate Gradient (CGNE)" <<std::endl;
|
|
||||||
Dw.Mdag(src,tmp);
|
|
||||||
src=tmp;
|
|
||||||
MdagMLinearOperator<WilsonFermionD,LatticeFermionD> HermOp(Dw);
|
|
||||||
ConjugateGradient<LatticeFermionD> CG(1.0e-10,10000);
|
|
||||||
CG(HermOp,src,result);
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
std::cout << " Taking difference" <<std::endl;
|
|
||||||
std::cout << "Dw result "<<norm2(result)<<std::endl;
|
|
||||||
std::cout << "Dw ref "<<norm2(ref)<<std::endl;
|
|
||||||
|
|
||||||
diff = ref - result;
|
|
||||||
std::cout << "result - ref "<<norm2(diff)<<std::endl;
|
|
||||||
|
|
||||||
DumpSliceNorm("Slice Norm Solution ",result,Nd-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
//Gauge invariance test
|
|
||||||
////////////////////////////////////////////////////
|
|
||||||
{
|
|
||||||
std::cout<<"****************************************"<<std::endl;
|
|
||||||
std::cout << "Gauge invariance test \n";
|
|
||||||
std::cout<<"****************************************"<<std::endl;
|
|
||||||
LatticeGaugeField U_GT(&GRID); // Gauge transformed field
|
|
||||||
LatticeColourMatrix g(&GRID); // local Gauge xform matrix
|
|
||||||
U_GT = Umu;
|
|
||||||
// Make a random xform to teh gauge field
|
|
||||||
SU<Nc>::RandomGaugeTransform(pRNG,U_GT,g); // Unit gauge
|
|
||||||
|
|
||||||
LatticeFermionD src(&GRID);
|
|
||||||
LatticeFermionD tmp(&GRID);
|
|
||||||
LatticeFermionD ref(&GRID);
|
|
||||||
LatticeFermionD diff(&GRID);
|
|
||||||
|
|
||||||
// could loop over colors
|
|
||||||
src=Zero();
|
|
||||||
Coordinate point(4,0); // 0,0,0,0
|
|
||||||
SpinColourVectorD ferm;
|
|
||||||
ferm=Zero();
|
|
||||||
ferm()(0)(0) = ComplexD(1.0);
|
|
||||||
pokeSite(ferm,src,point);
|
|
||||||
|
|
||||||
RealD mass=0.1;
|
|
||||||
WilsonFermionD Dw(U_GT,GRID,RBGRID,mass);
|
|
||||||
|
|
||||||
// Momentum space prop
|
|
||||||
std::cout << " Solving by FFT and Feynman rules" <<std::endl;
|
|
||||||
Dw.FreePropagator(src,ref,mass) ;
|
|
||||||
|
|
||||||
Gamma G5(Gamma::Algebra::Gamma5);
|
|
||||||
|
|
||||||
LatticeFermionD result(&GRID);
|
|
||||||
const int sdir=0;
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
// Conjugate gradient on normal equations system
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
std::cout << " Solving by Conjugate Gradient (CGNE)" <<std::endl;
|
|
||||||
Dw.Mdag(src,tmp);
|
|
||||||
src=tmp;
|
|
||||||
MdagMLinearOperator<WilsonFermionD,LatticeFermionD> HermOp(Dw);
|
|
||||||
ConjugateGradient<LatticeFermionD> CG(1.0e-10,10000);
|
|
||||||
CG(HermOp,src,result);
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////
|
|
||||||
std::cout << " Taking difference" <<std::endl;
|
|
||||||
std::cout << "Dw result "<<norm2(result)<<std::endl;
|
|
||||||
std::cout << "Dw ref "<<norm2(ref)<<std::endl;
|
|
||||||
|
|
||||||
diff = ref - result;
|
|
||||||
std::cout << "result - ref "<<norm2(diff)<<std::endl;
|
|
||||||
|
|
||||||
DumpSliceNorm("Slice Norm Solution ",result,Nd-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
Grid_finalize();
|
|
||||||
}
|
|
Reference in New Issue
Block a user