X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Ftt.h;h=e1bef323af5092474f07a96db48ff2d8fb8b4e9f;hp=a4ccd8be122d42d04603af957c81756ec224cde8;hb=2198cd0524574f0d9df8c0ec9aaf14ad8c94402b;hpb=fecefbb99cb0147f37d6895765a315f34c935786 diff --git a/src/tt.h b/src/tt.h index a4ccd8be..e1bef323 100644 --- a/src/tt.h +++ b/src/tt.h @@ -1,7 +1,8 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad + Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad + Copyright (C) 2015-2017 Marco Costalba, Joona Kiiski, Gary Linscott, 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 @@ -17,154 +18,104 @@ along with this program. If not, see . */ -#if !defined(TT_H_INCLUDED) +#ifndef TT_H_INCLUDED #define TT_H_INCLUDED -#include - -#include "move.h" +#include "misc.h" #include "types.h" - -/// The TTEntry is the class of transposition table entries -/// -/// A TTEntry needs 128 bits to be stored +/// TTEntry struct is the 10 bytes transposition table entry, defined as below: /// -/// bit 0-31: key -/// bit 32-63: data -/// bit 64-79: value -/// bit 80-95: depth -/// bit 96-111: static value -/// bit 112-127: margin of static value -/// -/// the 32 bits of the data field are so defined -/// -/// bit 0-15: move -/// bit 16-20: not used -/// bit 21-22: value type -/// bit 23-31: generation - -class TTEntry { - -public: - void save(uint32_t k, Value v, ValueType t, Depth d, Move m, int g, Value statV, Value statM) { - - key32 = (uint32_t)k; - move16 = (uint16_t)m; - valueType = (uint8_t)t; - generation8 = (uint8_t)g; - value16 = (int16_t)v; - depth16 = (int16_t)d; - staticValue = (int16_t)statV; - staticMargin = (int16_t)statM; +/// key 16 bit +/// move 16 bit +/// value 16 bit +/// eval value 16 bit +/// generation 6 bit +/// bound type 2 bit +/// depth 8 bit + +struct TTEntry { + + Move move() const { return (Move )move16; } + Value value() const { return (Value)value16; } + Value eval() const { return (Value)eval16; } + Depth depth() const { return (Depth)(depth8 * int(ONE_PLY)); } + Bound bound() const { return (Bound)(genBound8 & 0x3); } + + void save(Key k, Value v, Bound b, Depth d, Move m, Value ev, uint8_t g) { + + assert(d / ONE_PLY * ONE_PLY == d); + + // Preserve any existing move for the same position + if (m || (k >> 48) != key16) + move16 = (uint16_t)m; + + // Don't overwrite more valuable entries + if ( (k >> 48) != key16 + || d / ONE_PLY > depth8 - 4 + /* || g != (genBound8 & 0xFC) // Matching non-zero keys are already refreshed by probe() */ + || b == BOUND_EXACT) + { + key16 = (uint16_t)(k >> 48); + value16 = (int16_t)v; + eval16 = (int16_t)ev; + genBound8 = (uint8_t)(g | b); + depth8 = (int8_t)(d / ONE_PLY); + } } - void set_generation(int g) { generation8 = (uint8_t)g; } - - uint32_t key() const { return key32; } - Depth depth() const { return (Depth)depth16; } - Move move() const { return (Move)move16; } - Value value() const { return (Value)value16; } - ValueType type() const { return (ValueType)valueType; } - int generation() const { return (int)generation8; } - Value static_value() const { return (Value)staticValue; } - Value static_value_margin() const { return (Value)staticMargin; } private: - uint32_t key32; + friend class TranspositionTable; + + uint16_t key16; uint16_t move16; - uint8_t valueType, generation8; - int16_t value16, depth16, staticValue, staticMargin; + int16_t value16; + int16_t eval16; + uint8_t genBound8; + int8_t depth8; }; -/// This is the number of TTEntry slots for each cluster -const int ClusterSize = 4; - - -/// TTCluster consists of ClusterSize number of TTEntries. Size of TTCluster -/// must not be bigger than a cache line size. In case it is less, it should -/// be padded to guarantee always aligned accesses. - -struct TTCluster { - TTEntry data[ClusterSize]; -}; +/// A TranspositionTable consists of a power of 2 number of clusters and each +/// cluster consists of ClusterSize number of TTEntry. Each non-empty entry +/// contains information of exactly one position. The size of a cluster should +/// divide the size of a cache line size, to ensure that clusters never cross +/// cache lines. This ensures best cache performance, as the cacheline is +/// prefetched, as soon as possible. +class TranspositionTable { -/// The transposition table class. This is basically just a huge array containing -/// TTCluster objects, and a few methods for writing and reading entries. + static const int CacheLineSize = 64; + static const int ClusterSize = 3; -class TranspositionTable { + struct Cluster { + TTEntry entry[ClusterSize]; + char padding[2]; // Align to a divisor of the cache line size + }; - TranspositionTable(const TranspositionTable&); - TranspositionTable& operator=(const TranspositionTable&); + static_assert(CacheLineSize % sizeof(Cluster) == 0, "Cluster size incorrect"); public: - TranspositionTable(); - ~TranspositionTable(); - void set_size(size_t mbSize); + ~TranspositionTable() { free(mem); } + void new_search() { generation8 += 4; } // Lower 2 bits are used by Bound + uint8_t generation() const { return generation8; } + TTEntry* probe(const Key key, bool& found) const; + int hashfull() const; + void resize(size_t mbSize); void clear(); - void store(const Key posKey, Value v, ValueType type, Depth d, Move m, Value statV, Value kingD); - TTEntry* retrieve(const Key posKey) const; - void new_search(); - TTEntry* first_entry(const Key posKey) const; - void refresh(const TTEntry* tte) const; + + // The 32 lowest order bits of the key are used to get the index of the cluster + TTEntry* first_entry(const Key key) const { + return &table[(uint32_t(key) * uint64_t(clusterCount)) >> 32].entry[0]; + } private: - size_t size; - TTCluster* entries; - uint8_t generation; // Size must be not bigger then TTEntry::generation8 + size_t clusterCount; + Cluster* table; + void* mem; + uint8_t generation8; // Size must be not bigger than TTEntry::genBound8 }; extern TranspositionTable TT; - -/// TranspositionTable::first_entry() returns a pointer to the first entry of -/// a cluster given a position. The lowest order bits of the key are used to -/// get the index of the cluster. - -inline TTEntry* TranspositionTable::first_entry(const Key posKey) const { - - return entries[((uint32_t)posKey) & (size - 1)].data; -} - - -/// TranspositionTable::refresh() updates the 'generation' value of the TTEntry -/// to avoid aging. Normally called after a TT hit. - -inline void TranspositionTable::refresh(const TTEntry* tte) const { - - const_cast(tte)->set_generation(generation); -} - - -/// A simple fixed size hash table used to store pawns and material -/// configurations. It is basically just an array of Entry objects. -/// Without cluster concept or overwrite policy. - -template -struct SimpleHash { - - typedef SimpleHash Base; - - void init() { - - entries = new (std::nothrow) Entry[HashSize]; - if (!entries) - { - std::cerr << "Failed to allocate " << HashSize * sizeof(Entry) - << " bytes for material hash table." << std::endl; - exit(EXIT_FAILURE); - } - memset(entries, 0, HashSize * sizeof(Entry)); - } - - virtual ~SimpleHash() { delete [] entries; } - - Entry* find(Key key) const { return entries + ((uint32_t)key & (HashSize - 1)); } - void prefetch(Key key) const { ::prefetch((char*)find(key)); } - -protected: - Entry* entries; -}; - -#endif // !defined(TT_H_INCLUDED) +#endif // #ifndef TT_H_INCLUDED