X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Ftt.cpp;h=956184a92576089b14ced36fdd57285babaa944a;hb=6f6f59ea6a066eac291eac5ad0369cd2e0daf739;hp=f2313eabf1c72eee350de8cdda7038519944d222;hpb=cd4604b05c2d61928b26ab50c5864c36ef1d3785;p=stockfish diff --git a/src/tt.cpp b/src/tt.cpp index f2313eab..956184a9 100644 --- a/src/tt.cpp +++ b/src/tt.cpp @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2009 Marco Costalba + Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -23,17 +23,13 @@ //// #include -#include #include -#include -#include "movegen.h" +#include "evaluate.h" #include "tt.h" - -/// This is the number of TTEntry slots for each position -static const int ClusterSize = 5; - +// The main transposition table +TranspositionTable TT; //// //// Functions @@ -41,7 +37,7 @@ static const int ClusterSize = 5; TranspositionTable::TranspositionTable() { - size = writes = 0; + size = 0; entries = 0; generation = 0; } @@ -55,29 +51,26 @@ TranspositionTable::~TranspositionTable() { /// TranspositionTable::set_size sets the size of the transposition table, /// measured in megabytes. -void TranspositionTable::set_size(unsigned mbSize) { - - assert(mbSize >= 4 && mbSize <= 4096); +void TranspositionTable::set_size(size_t mbSize) { - unsigned newSize = 1024; + size_t newSize = 1024; // We store a cluster of ClusterSize number of TTEntry for each position // and newSize is the maximum number of storable positions. - while ((2 * newSize) * ClusterSize * (sizeof(TTEntry)) <= (mbSize << 20)) + while ((2 * newSize) * sizeof(TTCluster) <= (mbSize << 20)) newSize *= 2; if (newSize != size) { size = newSize; delete [] entries; - entries = new TTEntry[size * ClusterSize]; + entries = new TTCluster[size]; if (!entries) { std::cerr << "Failed to allocate " << mbSize << " MB for transposition table." << std::endl; Application::exit_with_failure(); } - clear(); } } @@ -89,7 +82,7 @@ void TranspositionTable::set_size(unsigned mbSize) { void TranspositionTable::clear() { - memset(entries, 0, size * ClusterSize * sizeof(TTEntry)); + memset(entries, 0, size * sizeof(TTCluster)); } @@ -103,8 +96,9 @@ void TranspositionTable::clear() { /// is bigger than the depth of t2. A TTEntry of type VALUE_TYPE_EVAL /// never replaces another entry for the same position. -void TranspositionTable::store(const Key posKey, Value v, ValueType t, Depth d, Move m) { +void TranspositionTable::store(const Key posKey, Value v, ValueType t, Depth d, Move m, Value statV, Value kingD) { + int c1, c2, c3; TTEntry *tte, *replace; uint32_t posKey32 = posKey >> 32; // Use the high 32 bits as key @@ -113,28 +107,25 @@ void TranspositionTable::store(const Key posKey, Value v, ValueType t, Depth d, { if (!tte->key() || tte->key() == posKey32) // empty or overwrite old { - // Do not overwrite when new type is VALUE_TYPE_EVAL - if (tte->key() && t == VALUE_TYPE_EVAL) - return; - + // Preserve any exsisting ttMove if (m == MOVE_NONE) m = tte->move(); - *tte = TTEntry(posKey32, v, t, d, m, generation); + tte->save(posKey32, v, t, d, m, generation, statV, kingD); return; } - else if (i == 0) // replace would be a no-op in this common case + + if (i == 0) // replace would be a no-op in this common case continue; - int c1 = (replace->generation() == generation ? 2 : 0); - int c2 = (tte->generation() == generation ? -2 : 0); - int c3 = (tte->depth() < replace->depth() ? 1 : 0); + c1 = (replace->generation() == generation ? 2 : 0); + c2 = (tte->generation() == generation ? -2 : 0); + c3 = (tte->depth() < replace->depth() ? 1 : 0); if (c1 + c2 + c3 > 0) replace = tte; } - *replace = TTEntry(posKey32, v, t, d, m, generation); - writes++; + replace->save(posKey32, v, t, d, m, generation, statV, kingD); } @@ -145,7 +136,7 @@ void TranspositionTable::store(const Key posKey, Value v, ValueType t, Depth d, TTEntry* TranspositionTable::retrieve(const Key posKey) const { uint32_t posKey32 = posKey >> 32; - TTEntry *tte = first_entry(posKey); + TTEntry* tte = first_entry(posKey); for (int i = 0; i < ClusterSize; i++, tte++) if (tte->key() == posKey32) @@ -154,26 +145,6 @@ TTEntry* TranspositionTable::retrieve(const Key posKey) const { return NULL; } -/// TranspositionTable::prefetch looks up the current position in the -/// transposition table and load it in L1/L2 cache. This is a non -/// blocking function and do not stalls the CPU waiting for data -/// to be loaded from RAM, that can be very slow. When we will -/// subsequently call retrieve() the TT data will be already -/// quickly accessible in L1/l2 CPU cache. - -void TranspositionTable::prefetch(const Key posKey) const { - - _mm_prefetch((char*)first_entry(posKey), _MM_HINT_T0); -} - -/// TranspositionTable::first_entry returns a pointer to the first -/// entry of a cluster given a position. The low 32 bits of the key -/// are used to get the index in the table. - -inline TTEntry* TranspositionTable::first_entry(const Key posKey) const { - - return entries + ((uint32_t(posKey) & (size - 1)) * ClusterSize); -} /// TranspositionTable::new_search() is called at the beginning of every new /// search. It increments the "generation" variable, which is used to @@ -181,68 +152,5 @@ inline TTEntry* TranspositionTable::first_entry(const Key posKey) const { /// entries from the current search. void TranspositionTable::new_search() { - generation++; - writes = 0; -} - - -/// TranspositionTable::insert_pv() is called at the end of a search -/// iteration, and inserts the PV back into the PV. This makes sure -/// the old PV moves are searched first, even if the old TT entries -/// have been overwritten. - -void TranspositionTable::insert_pv(const Position& pos, Move pv[]) { - - StateInfo st; - Position p(pos); - - for (int i = 0; pv[i] != MOVE_NONE; i++) - { - store(p.get_key(), VALUE_NONE, VALUE_TYPE_NONE, Depth(-127*OnePly), pv[i]); - p.do_move(pv[i], st); - } -} - - -/// TranspositionTable::extract_pv() extends a PV by adding moves from the -/// transposition table at the end. This should ensure that the PV is almost -/// always at least two plies long, which is important, because otherwise we -/// will often get single-move PVs when the search stops while failing high, -/// and a single-move PV means that we don't have a ponder move. - -void TranspositionTable::extract_pv(const Position& pos, Move pv[]) { - - int ply; - Position p(pos); - StateInfo st[100]; - - for (ply = 0; pv[ply] != MOVE_NONE; ply++) - p.do_move(pv[ply], st[ply]); - - bool stop; - const TTEntry* tte; - for (stop = false, tte = retrieve(p.get_key()); - tte && tte->move() != MOVE_NONE && !stop; - tte = retrieve(p.get_key()), ply++) - { - if (!move_is_legal(p, tte->move())) - break; - pv[ply] = tte->move(); - p.do_move(pv[ply], st[ply]); - for (int j = 0; j < ply; j++) - if (st[j].key == p.get_key()) stop = true; - } - pv[ply] = MOVE_NONE; -} - - -/// TranspositionTable::full() returns the permill of all transposition table -/// entries which have received at least one write during the current search. -/// It is used to display the "info hashfull ..." information in UCI. - -int TranspositionTable::full() const { - - double N = double(size) * ClusterSize; - return int(1000 * (1 - exp(writes * log(1.0 - 1.0/N)))); }