Compare commits
3 Commits
3127ac2ce5
...
9dd4c8d0bb
Author | SHA1 | Date | |
---|---|---|---|
9dd4c8d0bb | |||
81c887b27f | |||
540711e1ce |
9
.clang-format
Normal file
9
.clang-format
Normal file
@ -0,0 +1,9 @@
|
||||
BasedOnStyle: LLVM
|
||||
AlwaysBreakTemplateDeclarations: "Yes"
|
||||
AlignEscapedNewlines: Left
|
||||
BreakBeforeBraces: Allman
|
||||
FixNamespaceComments: "true"
|
||||
UseTab: Never
|
||||
IndentWidth: 2
|
||||
TabWidth: 2
|
||||
ColumnLimit: 100
|
43
RbcUkqcd.hpp
43
RbcUkqcd.hpp
@ -48,6 +48,10 @@ struct RbcUkqcd
|
||||
const std::string gaugeTransform,
|
||||
const std::string eigenpackPath,
|
||||
const double residual = 1.0e-8);
|
||||
static inline void addM0StrangeSolver(Application &app, const std::string solverName,
|
||||
const std::string gaugeName,
|
||||
const std::string gaugeTransform,
|
||||
const double residual = 1.0e-8);
|
||||
};
|
||||
|
||||
// Implementations /////////////////////////////////////////////////////////////////////////////////
|
||||
@ -113,4 +117,43 @@ void RbcUkqcd::addM0LightLCDSolver(Application &app, const std::string solverNam
|
||||
app.createModule<MSolver::MixedPrecisionRBPrecCGBatched>(solverName, solverPar);
|
||||
}
|
||||
|
||||
void RbcUkqcd::addM0StrangeSolver(Application &app, const std::string solverName,
|
||||
const std::string gaugeName, const std::string gaugeTransform,
|
||||
const double residual)
|
||||
{
|
||||
const std::string prefix = solverName;
|
||||
|
||||
// Gauge field FP32 cast
|
||||
MUtilities::GaugeSinglePrecisionCast::Par gaugeCastPar;
|
||||
|
||||
gaugeCastPar.field = gaugeName;
|
||||
app.createModule<MUtilities::GaugeSinglePrecisionCast>(prefix + "_gauge_fp32", gaugeCastPar);
|
||||
|
||||
// Scaled DWF action + FP32 version
|
||||
MAction::ScaledDWF::Par actionPar;
|
||||
|
||||
actionPar.gauge = gaugeName;
|
||||
actionPar.Ls = RbcUkqcd::m0LCDPar.Ls;
|
||||
actionPar.M5 = RbcUkqcd::m0LCDPar.M5;
|
||||
actionPar.mass = RbcUkqcd::m0LCDPar.ms;
|
||||
actionPar.scale = RbcUkqcd::m0LCDPar.scale;
|
||||
actionPar.boundary = "1 1 1 1";
|
||||
actionPar.twist = "0. 0. 0. 0.";
|
||||
app.createModule<MAction::ScaledDWF>(prefix + "_dwf", actionPar);
|
||||
actionPar.gauge = prefix + "_gauge_fp32";
|
||||
app.createModule<MAction::ScaledDWFF>(prefix + "_dwf_fp32", actionPar);
|
||||
|
||||
// Batched mixed-precision red-black preconditionned CG
|
||||
MSolver::MixedPrecisionRBPrecCG::Par solverPar;
|
||||
|
||||
solverPar.innerAction = prefix + "_dwf_fp32";
|
||||
solverPar.outerAction = prefix + "_dwf";
|
||||
solverPar.maxInnerIteration = 30000;
|
||||
solverPar.maxOuterIteration = 100;
|
||||
solverPar.residual = residual;
|
||||
solverPar.innerGuesser = "";
|
||||
solverPar.outerGuesser = "";
|
||||
app.createModule<MSolver::MixedPrecisionRBPrecCG>(solverName, solverPar);
|
||||
}
|
||||
|
||||
} // namespace hadpresets
|
||||
|
66
Test.hpp
Normal file
66
Test.hpp
Normal file
@ -0,0 +1,66 @@
|
||||
/*
|
||||
* Copyright (C) 2023
|
||||
*
|
||||
* Author: Antonin Portelli <antonin.portelli@me.com>
|
||||
* Elements based on production templates from Raoul Hodgson
|
||||
*
|
||||
* Hadrons 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.
|
||||
*
|
||||
* Hadrons 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 Hadrons. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* See the full license in the file "LICENSE" in the top level distribution
|
||||
* directory.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <Hadrons/Application.hpp>
|
||||
#include <Hadrons/Modules.hpp>
|
||||
|
||||
namespace hadpresets
|
||||
{
|
||||
using namespace Grid::Hadrons;
|
||||
|
||||
struct Test
|
||||
{
|
||||
static inline void addTestDwfSolver(Application &app, const std::string solverName,
|
||||
const std::string gaugeName, const double mass = 0.1,
|
||||
const double residual = 1.0e-8);
|
||||
};
|
||||
|
||||
void Test::addTestDwfSolver(Application &app, const std::string solverName,
|
||||
const std::string gaugeName, const double mass, const double residual)
|
||||
{
|
||||
const std::string prefix = solverName;
|
||||
|
||||
// DWF action
|
||||
MAction::DWF::Par actionPar;
|
||||
|
||||
actionPar.gauge = gaugeName;
|
||||
actionPar.Ls = 8;
|
||||
actionPar.M5 = 1.8;
|
||||
actionPar.mass = mass;
|
||||
actionPar.boundary = "1 1 1 -1";
|
||||
actionPar.twist = "0. 0. 0. 0.";
|
||||
app.createModule<MAction::DWF>(prefix + "_dwf", actionPar);
|
||||
|
||||
// solver
|
||||
MSolver::RBPrecCG::Par solverPar;
|
||||
|
||||
solverPar.action = prefix + "_dwf";
|
||||
solverPar.guesser = "";
|
||||
solverPar.maxIteration = 10000;
|
||||
solverPar.residual = residual;
|
||||
app.createModule<MSolver::RBPrecCG>(solverName, solverPar);
|
||||
}
|
||||
|
||||
} // namespace hadpresets
|
Reference in New Issue
Block a user