Repository navigation
Expand file tree
/
Copy pathGTP.h
More file actions
159 lines (136 loc) · 4.96 KB
/
Copy pathGTP.h
File metadata and controls
159 lines (136 loc) · 4.96 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
/*
This file is part of Leela Zero.
Copyright (C) 2017-2019 Gian-Carlo Pascutto and contributors
Leela Zero is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Leela Zero is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Leela Zero. If not, see <http://www.gnu.org/licenses/>.
Additional permission under GNU GPL version 3 section 7
If you modify this Program, or any covered work, by linking or
combining it with NVIDIA Corporation's libraries from the
NVIDIA CUDA Toolkit and/or the NVIDIA CUDA Deep Neural
Network library and/or the NVIDIA TensorRT inference library
(or a modified version of those libraries), containing parts covered
by the terms of the respective license agreement, the licensors of
this Program grant you additional permission to convey the resulting
work.
*/
#ifndef GTP_H_INCLUDED
#define GTP_H_INCLUDED
#include "config.h"
#include <cstdio>
#include <string>
#include <vector>
#include "Network.h"
#include "GameState.h"
#include "UCTSearch.h"
struct MoveToAvoid {
int color;
size_t until_move;
int vertex;
MoveToAvoid(int color, size_t until_move, int vertex)
: color(color), until_move(until_move), vertex(vertex)
{}
bool operator==(const MoveToAvoid other) const {
return color == other.color &&
until_move == other.until_move && vertex == other.vertex;
}
};
class AnalyzeTags {
friend class LeelaTest;
public:
AnalyzeTags() = default;
AnalyzeTags(std::istringstream& cmdstream, const GameState& game);
void add_move_to_avoid(int color, int vertex, size_t until_move);
void add_move_to_allow(int color, int vertex, size_t until_move);
int interval_centis() const;
int invalid() const;
int who() const;
size_t post_move_count() const;
bool is_to_avoid(int color, int vertex, size_t movenum) const;
bool has_move_restrictions() const;
private:
bool m_invalid{true};
std::vector<MoveToAvoid> m_moves_to_avoid, m_moves_to_allow;
int m_interval_centis{0};
int m_who{FastBoard::INVAL};
size_t m_min_moves{0};
};
extern bool cfg_gtp_mode;
extern bool cfg_allow_pondering;
extern unsigned int cfg_num_threads;
extern unsigned int cfg_batch_size;
extern int cfg_max_playouts;
extern int cfg_max_visits;
extern size_t cfg_max_memory;
extern size_t cfg_max_tree_size;
extern int cfg_max_cache_ratio_percent;
extern TimeManagement::enabled_t cfg_timemanage;
extern int cfg_lagbuffer_cs;
extern int cfg_resignpct;
extern int cfg_noise;
extern int cfg_random_cnt;
extern int cfg_random_min_visits;
extern float cfg_random_temp;
extern std::uint64_t cfg_rng_seed;
extern bool cfg_dumbpass;
#ifdef USE_OPENCL
extern std::vector<int> cfg_gpus;
extern bool cfg_sgemm_exhaustive;
extern bool cfg_tune_only;
#ifdef USE_HALF
enum class precision_t {
AUTO, SINGLE, HALF
};
extern precision_t cfg_precision;
#endif
#endif
extern float cfg_puct;
extern float cfg_logpuct;
extern float cfg_logconst;
extern float cfg_softmax_temp;
extern float cfg_fpu_reduction;
extern float cfg_fpu_root_reduction;
extern std::string cfg_logfile;
extern std::string cfg_weightsfile;
extern FILE* cfg_logfile_handle;
extern bool cfg_quiet;
extern std::string cfg_options_str;
extern bool cfg_benchmark;
extern bool cfg_cpu_only;
extern AnalyzeTags cfg_analyze_tags;
static constexpr size_t MiB = 1024LL * 1024LL;
/*
A list of all valid GTP2 commands is defined here:
https://www.lysator.liu.se/~gunnar/gtp/gtp2-spec-draft2/gtp2-spec.html
GTP is meant to be used between programs. It's not a human interface.
*/
class GTP {
public:
static std::unique_ptr<Network> s_network;
static void initialize(std::unique_ptr<Network>&& network);
static void execute(GameState & game, const std::string& xinput);
static void setup_default_parameters();
private:
static constexpr int GTP_VERSION = 2;
static std::string get_life_list(const GameState & game, bool live);
static const std::string s_commands[];
static const std::string s_options[];
static std::pair<std::string, std::string> parse_option(
std::istringstream& is);
static std::pair<bool, std::string> set_max_memory(
size_t max_memory, int cache_size_ratio_percent);
static void execute_setoption(UCTSearch& search,
int id, const std::string& command);
// Memory estimation helpers
static size_t get_base_memory();
static size_t add_overhead(size_t s) { return s * 11LL / 10LL; }
static size_t remove_overhead(size_t s) { return s * 10LL / 11LL; }
};
#endif