github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-07 19:40:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

llmodel: constify LLModel::threadCount()

Browse Source
This commit is contained in:
Juuso Alasuutari
2023-05-21 23:45:29 +03:00
committed by AT
parent efd39b0d73
commit 81fdc28e58
9 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gpt4all-backend/gptj.cpp
View File

@@ -867,7 +867,8 @@ void GPTJ::setThreadCount(int32_t n_threads) {
d_ptr->n_threads = n_threads;
}
int32_t GPTJ::threadCount() {
int32_t GPTJ::threadCount() const
{
return d_ptr->n_threads;
}

2
gpt4all-backend/gptj.h
View File

@@ -23,7 +23,7 @@ public:
std::function<bool(bool)> recalculateCallback,
PromptContext &ctx) override;
void setThreadCount(int32_t n_threads) override;
int32_t threadCount() override;
int32_t threadCount() const override;
protected:
void recalculateContext(PromptContext &promptCtx,

3
gpt4all-backend/llamamodel.cpp
View File

@@ -75,7 +75,8 @@ void LLamaModel::setThreadCount(int32_t n_threads) {
d_ptr->n_threads = n_threads;
}
int32_t LLamaModel::threadCount() {
int32_t LLamaModel::threadCount() const
{
return d_ptr->n_threads;
}

2
gpt4all-backend/llamamodel.h
View File

@@ -23,7 +23,7 @@ public:
std::function<bool(bool)> recalculateCallback,
PromptContext &ctx) override;
void setThreadCount(int32_t n_threads) override;
int32_t threadCount() override;
int32_t threadCount() const override;
protected:
void recalculateContext(PromptContext &promptCtx,

2
gpt4all-backend/llmodel.h
View File

@@ -37,7 +37,7 @@ public:
std::function<bool(bool)> recalculateCallback,
PromptContext &ctx) = 0;
virtual void setThreadCount(int32_t n_threads) {}
virtual int32_t threadCount() { return 1; }
virtual int32_t threadCount() const { return 1; }
protected:
virtual void recalculateContext(PromptContext &promptCtx,

3
gpt4all-backend/mpt.cpp
View File

@@ -792,7 +792,8 @@ void MPT::setThreadCount(int32_t n_threads) {
d_ptr->n_threads = n_threads;
}
int32_t MPT::threadCount() {
int32_t MPT::threadCount() const
{
return d_ptr->n_threads;
}

2
gpt4all-backend/mpt.h
View File

@@ -23,7 +23,7 @@ public:
std::function<bool(bool)> recalculateCallback,
PromptContext &ctx) override;
void setThreadCount(int32_t n_threads) override;
int32_t threadCount() override;
int32_t threadCount() const override;
protected:
void recalculateContext(PromptContext &promptCtx,