JonKazama-Hellion/Craftimizer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Use rqrt in child score calculation

Browse Source 
Changes sqrt -> rsqrt and div->mul. Reduces latency by 23 -> 8 cycles.
This commit is contained in:
Asriel Camora
2023-07-04 08:36:41 +02:00
parent 76853e2f0d
commit e46d1b20fa
2 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Solver/Crafty/Solver.cs
+6 -3
View File
@@ -101,7 +101,7 @@ public class Solver
var length = children.Length;
var vecLength = Vector<float>.Count;
var C = Config.ExplorationConstant * MathF.Log(parentVisits);
var C = MathF.Sqrt(Config.ExplorationConstant * MathF.Log(parentVisits));
var w = Config.MaxScoreWeightingConstant;
var W = 1f - w;
var CVector = new Vector<float>(C);
@@ -126,8 +126,11 @@ public class Solver
maxScores[j] = node.MaxScore;
}
var exploitation = (W * (new Vector<float>(scoreSums) / new Vector<float>(visits))) + (w * new Vector<float>(maxScores));
var exploration = Vector.SquareRoot(CVector / new Vector<float>(visits));
var s = new Vector<float>(scoreSums);
var m = new Vector<float>(maxScores);
var v = new Vector<float>(visits);
var exploitation = (W * (s / v)) + (w * m);
var exploration = CVector * Intrinsics.ReciprocalSqrt(v);
var evalScores = exploitation + exploration;
var idx = Intrinsics.HMaxIndex(evalScores, iterCount);