diff --git a/bestla/bestla/bestla_parallel.h b/bestla/bestla/bestla_parallel.h index 6d9ddc1c0..04a310a5e 100644 --- a/bestla/bestla/bestla_parallel.h +++ b/bestla/bestla/bestla_parallel.h @@ -634,7 +634,7 @@ class SchedulerKBlockS : public SchedulerBase<_GemmCore_T> { } else { this->mBlock[1] = this->mThdSize[1]; } - size_t csize = static_cast(mBlock[0]) * mBlock[1] * this->mEleSize[2]; + size_t csize = static_cast(this->mBlock[0]) * this->mBlock[1] * this->mEleSize[2]; auto rawk = static_cast((valid_total - csize) / 2 / (this->mStep[0] * this->mEleSize[0] + float(CorSize * (this->mStep[0] + this->mBlock[1])) / this->mKBlock +