Skip to content

Commit

Permalink
add naik derivative
Browse files Browse the repository at this point in the history
  • Loading branch information
@jcosborn
jcosborn committed Dec 6, 2024
1 parent 476ef08 commit 5b97195
Show file tree
Hide file tree
Showing 4 changed files with 78 additions and 15 deletions.
13 changes: 7 additions & 6 deletions src/gauge/fat7l.nim
View file
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,9 @@ proc computeGenStaple(staple: auto, mu,nu: int, link: auto, coef: float,
#if(link!=gauge[mu]) QDP_discard_M(ts0);
#QDP_discard_M(ts2);

proc makeImpLinks*(info: var PerfInfo, fl: auto, gf: auto, coef: auto,
ll: auto, gfLong: auto, naik: auto) =
proc makeImpLinks*(fl: auto, gf: auto, coef: auto,
ll: auto, gfLong: auto, naik: auto,
info: var PerfInfo) =
tic()
type lcm = type(gf[0])
proc QDP_create_M(): lcm = result.new(gf[0].l)
Expand Down Expand Up @@ -168,8 +169,8 @@ proc makeImpLinks*(info: var PerfInfo, fl: auto, gf: auto, coef: auto,
info.flops += nflop * gf[0].l.localGeom.prod
info.secs += getElapsedTime()

proc makeImpLinks*(info: var PerfInfo, fl: auto, gf: auto, coef: auto) =
makeImpLinks(info, fl, gf, coef, fl, gf, 0.0)
proc makeImpLinks*(fl: auto, gf: auto, coef: auto, info: var PerfInfo) =
makeImpLinks(fl, gf, coef, fl, gf, 0.0, info)

when isMainModule:
import qex
Expand All @@ -196,8 +197,8 @@ when isMainModule:
var g3 = g

echo g.plaq
makeImpLinks(info, fl, g, coef)
makeImpLinks(info, fl, g, coef, ll, g3, naik)
makeImpLinks(fl, g, coef, info)
makeImpLinks(fl, g, coef, ll, g3, naik, info)
echo fl.plaq
echo ll.plaq
echo pow(1.0,4)/6.0
Expand Down
74 changes: 68 additions & 6 deletions src/gauge/fat7lderiv.nim
View file
Original file line number Diff line number Diff line change
Expand Up @@ -119,8 +119,9 @@ side_force(QDP_ColorMatrix *force, QDP_ColorMatrix *bot0, QDP_ColorMatrix *side0
}
]#

proc fat7lderiv(perf: var PerfInfo, gauge: auto, deriv: auto,
coef: Fat7lCoefs, mid: auto) =
proc fat7lDeriv(deriv: auto, gauge: auto, mid: auto, coef: Fat7lCoefs,
llderiv: auto, llgauge: auto, llmid: auto, naik: float,
perf: var PerfInfo) =
tic()
var nflops = 0
let coefL = coef.lepage
Expand Down Expand Up @@ -287,9 +288,49 @@ proc fat7lderiv(perf: var PerfInfo, gauge: auto, deriv: auto,
deriv[mu] += fixL * stpl5
nflops += 4*(3*(2*198+3*18)+198+216+36)

if naik!=0.0:
for mu in 0..3:
# force += mid * Umumu' * Umu'
# force -= U-mu' * mid-mu * Umu'
# force += U-mu' * U-mu-mu' * mid-mu-mu
#QDP_M_eq_M(Uf, U, QDP_all);
#QDP_M_eq_sM(Umu, Uf, QDP_neighbor[mu], QDP_forward, QDP_all);
discard s1a[mu] ^* llgauge[mu]
#QDP_M_eq_Ma_times_M(Umid, Uf, mid, QDP_all);
tm1 := llgauge[mu].adj * llmid[mu]
#QDP_M_eq_sM(Umidbmu, Umid, QDP_neighbor[mu], QDP_backward, QDP_all);
discard sm1[mu] ^* tm1
#QDP_M_eq_Ma_times_Ma(UmuU, Umu, Uf, QDP_all);
tm2 := s1a[mu].field.adj * llgauge[mu].adj
#QDP_M_eq_sM(UmuUs, UmuU, QDP_neighbor[mu], QDP_forward, QDP_all);
discard s1b[mu] ^* tm2
#QDP_M_eq_Ma_times_M(f3b, Uf, Umidbmu, QDP_all);
llderiv[mu] += naik*(sm1[mu].field*s1a[mu].field.adj)
tm1 := llgauge[mu].adj * sm1[mu].field
#QDP_M_eq_sM(f3, f3b, QDP_neighbor[mu], QDP_backward, QDP_all);
discard sm1[mu] ^* tm1
#QDP_M_eq_M_times_M(f, mid, UmuUs, QDP_all);
#QDP_discard_M(UmuUs);
#QDP_M_peq_M_times_Ma(f, Umidbmu, Umu, QDP_all);
#QDP_discard_M(Umidbmu);
#QDP_discard_M(Umu);
#QDP_M_peq_M(f, f3, QDP_all);
#QDP_discard_M(f3);
#QDP_M_peq_r_times_M(deriv[mu], &coefN, f, QDP_all);
llderiv[mu] += naik*(llmid[mu]*s1b[mu].field+sm1[mu].field)

#info->final_sec = QOP_time() - dtime;
#info->final_flop = nflops*QDP_sites_on_node;
#info->status = QOP_SUCCESS;
toc()
inc perf.count
perf.flops += nflops * gauge[0].l.localGeom.prod
perf.secs += getElapsedTime()

proc fat7lDeriv(deriv: auto, gauge: auto, mid: auto, coef: Fat7lCoefs,
perf: var PerfInfo) =
fat7lDeriv(deriv, gauge, mid, coef, deriv, gauge, mid, 0.0, perf)


when isMainModule:
import qex, physics/qcdTypes
Expand All @@ -308,23 +349,27 @@ when isMainModule:
coef.fiveStaple = 1.0
coef.sevenStaple = 1.0
coef.lepage = 1.0
var naik = 1.0

var fl = lo.newGauge()
var fl2 = lo.newGauge()
var ll = lo.newGauge()
var ll2 = lo.newGauge()
var dfl = lo.newGauge()
var g2 = lo.newGauge()
var dg = lo.newGauge()
var fd = lo.newGauge()
var ld = lo.newGauge()
for mu in 0..<dg.len:
dg[mu] := 0.00001 * g[mu]
g2[mu] := g[mu] + dg[mu]
fd[mu] := 0

echo g.plaq
echo g2.plaq
makeImpLinks(info, fl, g, coef)
makeImpLinks(fl, g, coef, info)
echo info
makeImpLinks(info, fl2, g2, coef)
makeImpLinks(fl2, g2, coef, info)
echo info
echo fl.plaq
echo fl2.plaq
Expand All @@ -335,13 +380,30 @@ when isMainModule:
#echo pow(1.0+6.0+6.0*4.0+6.0*4.0*2.0,4)/6.0
#echo pow(1.0+6.0+6.0*4.0+6.0*4.0*2.0+6.0,4)/6.0

fat7lderiv(info, g, fd, coef, dg)
fat7lderiv(fd, g, dg, coef, info)
echo info
for mu in 0..3:
dfl[mu] := fl2[mu] - fl[mu]
#echo dfl[mu].norm2
#echo fd[mu].norm2
dfl[mu] -= fd[mu]
echo dfl[mu].norm2
echo fd[mu].norm2

for mu in 0..<dg.len:
fd[mu] := 0
ld[mu] := 0
makeImpLinks(fl, g, coef, ll, g, naik, info)
makeImpLinks(fl2, g2, coef, ll2, g2, naik, info)
fat7lderiv(fd, g, dg, coef, ld, g, dg, naik, info)
echo info
for mu in 0..3:
dfl[mu] := fl2[mu] - fl[mu]
#echo dfl[mu].norm2
#echo fd[mu].norm2
dfl[mu] -= fd[mu]
echo dfl[mu].norm2
dfl[mu] := ll2[mu] - ll[mu]
dfl[mu] -= ld[mu]
echo dfl[mu].norm2

qexFinalize()
2 changes: 1 addition & 1 deletion src/physics/asqtadLinks.nim
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ proc `$`*(c: AsqtadCoefs): string =

proc smear*(c: AsqtadCoefs, g: any, fl,ll: any) =
var info: PerfInfo
makeImpLinks(info, fl, g, c.fat7, ll, g, c.naik)
makeImpLinks(fl, g, c.fat7, ll, g, c.naik, info)

when isMainModule:
import qex
Expand Down
4 changes: 2 additions & 2 deletions src/physics/hisqLinks.nim
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,11 +32,11 @@ proc `$`*(c: HisqCoefs): string =
proc smear*(c: HisqCoefs, g: auto, fl,ll: auto, t1,t2: auto) =
mixin projectU
var info: PerfInfo
makeImpLinks(info, t1, g, c.fat7first)
makeImpLinks(t1, g, c.fat7first, info)
for mu in 0..<4:
for i in t2[mu]:
projectU(t2[mu][i], t1[mu][i])
makeImpLinks(info, fl, t2, c.fat7second, ll, t2, c.naik)
makeImpLinks(fl, t2, c.fat7second, ll, t2, c.naik, info)
proc smear*(c: HisqCoefs, g: auto, fl,ll: auto) =
var t1 = fl.newOneOf
var t2 = fl.newOneOf
Expand Down

0 comments on commit 5b97195

Please sign in to comment.