mirror of
https://github.com/sockspls/badfish
synced 2025-04-29 16:23:09 +00:00
Material: lockless per-thread maps
Adds a good bunch of code but should be faster and scalable because is lockless. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
8b57416ace
commit
899b9455d6
3 changed files with 109 additions and 86 deletions
187
src/material.cpp
187
src/material.cpp
|
@ -25,7 +25,6 @@
|
||||||
#include <cassert>
|
#include <cassert>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
#include "lock.h"
|
|
||||||
#include "material.h"
|
#include "material.h"
|
||||||
|
|
||||||
|
|
||||||
|
@ -46,54 +45,26 @@ namespace {
|
||||||
ScalingFunction* fun;
|
ScalingFunction* fun;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
////
|
||||||
|
//// Classes
|
||||||
|
////
|
||||||
|
|
||||||
|
class EndgameFunctions {
|
||||||
|
|
||||||
|
public:
|
||||||
|
EndgameFunctions();
|
||||||
|
EndgameEvaluationFunction* getEEF(Key key);
|
||||||
|
ScalingInfo getESF(Key key);
|
||||||
|
|
||||||
|
private:
|
||||||
|
void add(Key k, EndgameEvaluationFunction* f);
|
||||||
|
void add(Key k, Color c, ScalingFunction* f);
|
||||||
|
|
||||||
std::map<Key, EndgameEvaluationFunction*> EEFmap;
|
std::map<Key, EndgameEvaluationFunction*> EEFmap;
|
||||||
std::map<Key, ScalingInfo> ESFmap;
|
std::map<Key, ScalingInfo> ESFmap;
|
||||||
|
};
|
||||||
Lock EEFmapLock;
|
|
||||||
Lock ESFmapLock;
|
|
||||||
|
|
||||||
void add(Key k, EndgameEvaluationFunction* f) {
|
|
||||||
|
|
||||||
EEFmap.insert(std::pair<Key, EndgameEvaluationFunction*>(k, f));
|
|
||||||
}
|
|
||||||
|
|
||||||
void add(Key k, Color c, ScalingFunction* f) {
|
|
||||||
|
|
||||||
ScalingInfo s = {c, f};
|
|
||||||
ESFmap.insert(std::pair<Key, ScalingInfo>(k, s));
|
|
||||||
}
|
|
||||||
|
|
||||||
// STL map are not guaranteed to be thread safe even
|
|
||||||
// for read-access so we need this two helpers to access them.
|
|
||||||
EndgameEvaluationFunction* getEEF(Key key) {
|
|
||||||
|
|
||||||
EndgameEvaluationFunction* f = NULL;
|
|
||||||
|
|
||||||
lock_grab(&EEFmapLock);
|
|
||||||
|
|
||||||
std::map<Key, EndgameEvaluationFunction*>::iterator it(EEFmap.find(key));
|
|
||||||
if (it != EEFmap.end())
|
|
||||||
f = it->second;
|
|
||||||
|
|
||||||
lock_release(&EEFmapLock);
|
|
||||||
return f;
|
|
||||||
}
|
|
||||||
|
|
||||||
ScalingInfo getESF(Key key) {
|
|
||||||
|
|
||||||
ScalingInfo si = {WHITE, NULL};
|
|
||||||
|
|
||||||
lock_grab(&ESFmapLock);
|
|
||||||
|
|
||||||
std::map<Key, ScalingInfo>::iterator it(ESFmap.find(key));
|
|
||||||
if (it != ESFmap.end())
|
|
||||||
si = it->second;
|
|
||||||
|
|
||||||
lock_release(&ESFmapLock);
|
|
||||||
return si;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
////
|
////
|
||||||
|
@ -106,45 +77,11 @@ namespace {
|
||||||
|
|
||||||
void MaterialInfo::init() {
|
void MaterialInfo::init() {
|
||||||
|
|
||||||
// Initialize std::map access locks
|
|
||||||
lock_init(&EEFmapLock, NULL);
|
|
||||||
lock_init(&ESFmapLock, NULL);
|
|
||||||
|
|
||||||
typedef Key ZM[2][8][16];
|
typedef Key ZM[2][8][16];
|
||||||
const ZM& z = Position::zobMaterial;
|
const ZM& z = Position::zobMaterial;
|
||||||
|
|
||||||
static const Color W = WHITE;
|
KNNKMaterialKey = z[WHITE][KNIGHT][1] ^ z[WHITE][KNIGHT][2];
|
||||||
static const Color B = BLACK;
|
KKNNMaterialKey = z[BLACK][KNIGHT][1] ^ z[BLACK][KNIGHT][2];
|
||||||
|
|
||||||
KNNKMaterialKey = z[W][KNIGHT][1] ^ z[W][KNIGHT][2];
|
|
||||||
KKNNMaterialKey = z[B][KNIGHT][1] ^ z[B][KNIGHT][2];
|
|
||||||
|
|
||||||
add(z[W][PAWN][1], &EvaluateKPK);
|
|
||||||
add(z[B][PAWN][1], &EvaluateKKP);
|
|
||||||
|
|
||||||
add(z[W][BISHOP][1] ^ z[W][KNIGHT][1], &EvaluateKBNK);
|
|
||||||
add(z[B][BISHOP][1] ^ z[B][KNIGHT][1], &EvaluateKKBN);
|
|
||||||
add(z[W][ROOK][1] ^ z[B][PAWN][1], &EvaluateKRKP);
|
|
||||||
add(z[W][PAWN][1] ^ z[B][ROOK][1], &EvaluateKPKR);
|
|
||||||
add(z[W][ROOK][1] ^ z[B][BISHOP][1], &EvaluateKRKB);
|
|
||||||
add(z[W][BISHOP][1] ^ z[B][ROOK][1], &EvaluateKBKR);
|
|
||||||
add(z[W][ROOK][1] ^ z[B][KNIGHT][1], &EvaluateKRKN);
|
|
||||||
add(z[W][KNIGHT][1] ^ z[B][ROOK][1], &EvaluateKNKR);
|
|
||||||
add(z[W][QUEEN][1] ^ z[B][ROOK][1], &EvaluateKQKR);
|
|
||||||
add(z[W][ROOK][1] ^ z[B][QUEEN][1], &EvaluateKRKQ);
|
|
||||||
|
|
||||||
add(z[W][KNIGHT][1] ^ z[W][PAWN][1], W, &ScaleKNPK);
|
|
||||||
add(z[B][KNIGHT][1] ^ z[B][PAWN][1], B, &ScaleKKNP);
|
|
||||||
|
|
||||||
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[B][ROOK][1] , W, &ScaleKRPKR);
|
|
||||||
add(z[W][ROOK][1] ^ z[B][ROOK][1] ^ z[B][PAWN][1] , B, &ScaleKRKRP);
|
|
||||||
add(z[W][BISHOP][1] ^ z[W][PAWN][1] ^ z[B][BISHOP][1], W, &ScaleKBPKB);
|
|
||||||
add(z[W][BISHOP][1] ^ z[B][BISHOP][1] ^ z[B][PAWN][1] , B, &ScaleKBKBP);
|
|
||||||
add(z[W][BISHOP][1] ^ z[W][PAWN][1] ^ z[B][KNIGHT][1], W, &ScaleKBPKN);
|
|
||||||
add(z[W][KNIGHT][1] ^ z[B][BISHOP][1] ^ z[B][PAWN][1] , B, &ScaleKNKBP);
|
|
||||||
|
|
||||||
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[W][PAWN][2] ^ z[B][ROOK][1] ^ z[B][PAWN][1], W, &ScaleKRPPKRP);
|
|
||||||
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[B][ROOK][1] ^ z[B][PAWN][1] ^ z[B][PAWN][2], B, &ScaleKRPKRPP);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -154,7 +91,8 @@ MaterialInfoTable::MaterialInfoTable(unsigned int numOfEntries) {
|
||||||
|
|
||||||
size = numOfEntries;
|
size = numOfEntries;
|
||||||
entries = new MaterialInfo[size];
|
entries = new MaterialInfo[size];
|
||||||
if (!entries)
|
funcs = new EndgameFunctions();
|
||||||
|
if (!entries || !funcs)
|
||||||
{
|
{
|
||||||
std::cerr << "Failed to allocate " << (numOfEntries * sizeof(MaterialInfo))
|
std::cerr << "Failed to allocate " << (numOfEntries * sizeof(MaterialInfo))
|
||||||
<< " bytes for material hash table." << std::endl;
|
<< " bytes for material hash table." << std::endl;
|
||||||
|
@ -169,6 +107,7 @@ MaterialInfoTable::MaterialInfoTable(unsigned int numOfEntries) {
|
||||||
MaterialInfoTable::~MaterialInfoTable() {
|
MaterialInfoTable::~MaterialInfoTable() {
|
||||||
|
|
||||||
delete [] entries;
|
delete [] entries;
|
||||||
|
delete funcs;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -213,7 +152,7 @@ MaterialInfo *MaterialInfoTable::get_material_info(const Position& pos) {
|
||||||
|
|
||||||
// Let's look if we have a specialized evaluation function for this
|
// Let's look if we have a specialized evaluation function for this
|
||||||
// particular material configuration.
|
// particular material configuration.
|
||||||
if ((mi->evaluationFunction = getEEF(key)) != NULL)
|
if ((mi->evaluationFunction = funcs->getEEF(key)) != NULL)
|
||||||
return mi;
|
return mi;
|
||||||
|
|
||||||
else if ( pos.non_pawn_material(BLACK) == Value(0)
|
else if ( pos.non_pawn_material(BLACK) == Value(0)
|
||||||
|
@ -238,7 +177,7 @@ MaterialInfo *MaterialInfoTable::get_material_info(const Position& pos) {
|
||||||
// if we decide to add more special cases. We face problems when there
|
// if we decide to add more special cases. We face problems when there
|
||||||
// are several conflicting applicable scaling functions and we need to
|
// are several conflicting applicable scaling functions and we need to
|
||||||
// decide which one to use.
|
// decide which one to use.
|
||||||
ScalingInfo si = getESF(key);
|
ScalingInfo si = funcs->getESF(key);
|
||||||
if (si.fun != NULL)
|
if (si.fun != NULL)
|
||||||
{
|
{
|
||||||
mi->scalingFunction[si.col] = si.fun;
|
mi->scalingFunction[si.col] = si.fun;
|
||||||
|
@ -347,3 +286,79 @@ MaterialInfo *MaterialInfoTable::get_material_info(const Position& pos) {
|
||||||
mi->egValue = int16_t(egValue);
|
mi->egValue = int16_t(egValue);
|
||||||
return mi;
|
return mi;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/// EndgameFunctions members definition. This helper class is used to
|
||||||
|
/// store the maps of end game and scaling functions that MaterialInfoTable
|
||||||
|
/// will query for each key. The maps are constant, and are populated only
|
||||||
|
/// at construction. Being per thread avoids to use locks to access them.
|
||||||
|
|
||||||
|
EndgameFunctions::EndgameFunctions() {
|
||||||
|
|
||||||
|
typedef Key ZM[2][8][16];
|
||||||
|
const ZM& z = Position::zobMaterial;
|
||||||
|
|
||||||
|
static const Color W = WHITE;
|
||||||
|
static const Color B = BLACK;
|
||||||
|
|
||||||
|
KNNKMaterialKey = z[W][KNIGHT][1] ^ z[W][KNIGHT][2];
|
||||||
|
KKNNMaterialKey = z[B][KNIGHT][1] ^ z[B][KNIGHT][2];
|
||||||
|
|
||||||
|
add(z[W][PAWN][1], &EvaluateKPK);
|
||||||
|
add(z[B][PAWN][1], &EvaluateKKP);
|
||||||
|
|
||||||
|
add(z[W][BISHOP][1] ^ z[W][KNIGHT][1], &EvaluateKBNK);
|
||||||
|
add(z[B][BISHOP][1] ^ z[B][KNIGHT][1], &EvaluateKKBN);
|
||||||
|
add(z[W][ROOK][1] ^ z[B][PAWN][1], &EvaluateKRKP);
|
||||||
|
add(z[W][PAWN][1] ^ z[B][ROOK][1], &EvaluateKPKR);
|
||||||
|
add(z[W][ROOK][1] ^ z[B][BISHOP][1], &EvaluateKRKB);
|
||||||
|
add(z[W][BISHOP][1] ^ z[B][ROOK][1], &EvaluateKBKR);
|
||||||
|
add(z[W][ROOK][1] ^ z[B][KNIGHT][1], &EvaluateKRKN);
|
||||||
|
add(z[W][KNIGHT][1] ^ z[B][ROOK][1], &EvaluateKNKR);
|
||||||
|
add(z[W][QUEEN][1] ^ z[B][ROOK][1], &EvaluateKQKR);
|
||||||
|
add(z[W][ROOK][1] ^ z[B][QUEEN][1], &EvaluateKRKQ);
|
||||||
|
|
||||||
|
add(z[W][KNIGHT][1] ^ z[W][PAWN][1], W, &ScaleKNPK);
|
||||||
|
add(z[B][KNIGHT][1] ^ z[B][PAWN][1], B, &ScaleKKNP);
|
||||||
|
|
||||||
|
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[B][ROOK][1] , W, &ScaleKRPKR);
|
||||||
|
add(z[W][ROOK][1] ^ z[B][ROOK][1] ^ z[B][PAWN][1] , B, &ScaleKRKRP);
|
||||||
|
add(z[W][BISHOP][1] ^ z[W][PAWN][1] ^ z[B][BISHOP][1], W, &ScaleKBPKB);
|
||||||
|
add(z[W][BISHOP][1] ^ z[B][BISHOP][1] ^ z[B][PAWN][1] , B, &ScaleKBKBP);
|
||||||
|
add(z[W][BISHOP][1] ^ z[W][PAWN][1] ^ z[B][KNIGHT][1], W, &ScaleKBPKN);
|
||||||
|
add(z[W][KNIGHT][1] ^ z[B][BISHOP][1] ^ z[B][PAWN][1] , B, &ScaleKNKBP);
|
||||||
|
|
||||||
|
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[W][PAWN][2] ^ z[B][ROOK][1] ^ z[B][PAWN][1], W, &ScaleKRPPKRP);
|
||||||
|
add(z[W][ROOK][1] ^ z[W][PAWN][1] ^ z[B][ROOK][1] ^ z[B][PAWN][1] ^ z[B][PAWN][2], B, &ScaleKRPKRPP);
|
||||||
|
}
|
||||||
|
|
||||||
|
void EndgameFunctions::add(Key k, EndgameEvaluationFunction* f) {
|
||||||
|
|
||||||
|
EEFmap.insert(std::pair<Key, EndgameEvaluationFunction*>(k, f));
|
||||||
|
}
|
||||||
|
|
||||||
|
void EndgameFunctions::add(Key k, Color c, ScalingFunction* f) {
|
||||||
|
|
||||||
|
ScalingInfo s = {c, f};
|
||||||
|
ESFmap.insert(std::pair<Key, ScalingInfo>(k, s));
|
||||||
|
}
|
||||||
|
|
||||||
|
EndgameEvaluationFunction* EndgameFunctions::getEEF(Key key) {
|
||||||
|
|
||||||
|
EndgameEvaluationFunction* f = NULL;
|
||||||
|
std::map<Key, EndgameEvaluationFunction*>::iterator it(EEFmap.find(key));
|
||||||
|
if (it != EEFmap.end())
|
||||||
|
f = it->second;
|
||||||
|
|
||||||
|
return f;
|
||||||
|
}
|
||||||
|
|
||||||
|
ScalingInfo EndgameFunctions::getESF(Key key) {
|
||||||
|
|
||||||
|
ScalingInfo si = {WHITE, NULL};
|
||||||
|
std::map<Key, ScalingInfo>::iterator it(ESFmap.find(key));
|
||||||
|
if (it != ESFmap.end())
|
||||||
|
si = it->second;
|
||||||
|
|
||||||
|
return si;
|
||||||
|
}
|
||||||
|
|
|
@ -69,6 +69,12 @@ private:
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
/// Stores the endgame evaluation functions maps. Should be per thread
|
||||||
|
/// because STL is not thread safe and locks are expensive.
|
||||||
|
|
||||||
|
class EndgameFunctions;
|
||||||
|
|
||||||
|
|
||||||
/// The MaterialInfoTable class represents a pawn hash table. It is basically
|
/// The MaterialInfoTable class represents a pawn hash table. It is basically
|
||||||
/// just an array of MaterialInfo objects and a few methods for accessing these
|
/// just an array of MaterialInfo objects and a few methods for accessing these
|
||||||
/// objects. The most important method is get_material_info, which looks up a
|
/// objects. The most important method is get_material_info, which looks up a
|
||||||
|
@ -85,6 +91,7 @@ public:
|
||||||
private:
|
private:
|
||||||
unsigned size;
|
unsigned size;
|
||||||
MaterialInfo *entries;
|
MaterialInfo *entries;
|
||||||
|
EndgameFunctions* funcs;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -117,6 +117,7 @@ struct UndoInfo {
|
||||||
class Position {
|
class Position {
|
||||||
|
|
||||||
friend class MaterialInfo;
|
friend class MaterialInfo;
|
||||||
|
friend class EndgameFunctions;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
// Constructors
|
// Constructors
|
||||||
|
|
Loading…
Add table
Reference in a new issue