Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/amun/common/god.h')
-rw-r--r--src/amun/common/god.h6
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_;
};
}