diff options
Diffstat (limited to 'src/amun/common/god.h')
-rw-r--r-- | src/amun/common/god.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/amun/common/god.h b/src/amun/common/god.h index cd719ad9..2eceae80 100644 --- a/src/amun/common/god.h +++ b/src/amun/common/god.h @@ -88,6 +88,9 @@ class God { bool UseFusedSoftmax() const { return useFusedSoftmax_; } + bool UseTensorCores() const + { return useTensorCores_; } + private: void LoadScorers(); void LoadFiltering(); @@ -120,8 +123,7 @@ class God { std::unique_ptr<ThreadPool> pool_; - bool returnNBestList_; - bool useFusedSoftmax_; + bool returnNBestList_, useFusedSoftmax_, useTensorCores_; }; } |