55 lines
2.8 KiB
Diff
55 lines
2.8 KiB
Diff
diff --git a/cpp_thread_test/cpp_thread_safety_common.h b/cpp_thread_test/cpp_thread_safety_common.h
|
|
index 60ab5bb..8005369 100644
|
|
--- a/cpp_thread_test/cpp_thread_safety_common.h
|
|
+++ b/cpp_thread_test/cpp_thread_safety_common.h
|
|
@@ -5,6 +5,14 @@ inline void pauser(){
|
|
std::getline(std::cin, dummy);
|
|
}
|
|
|
|
+void FailIfThreadsAreZero(uint32_t numConcurrentThreads) {
|
|
+ if(numConcurrentThreads == 0) {
|
|
+ std::cout<<"ERROR: Invalid parameter 0 for number of concurrent calls into OpenBLAS!"<<std::endl;
|
|
+ std::cout<<"CBLAS DGEMV thread safety test FAILED!"<<std::endl;
|
|
+ exit(-1);
|
|
+ }
|
|
+}
|
|
+
|
|
void FillMatrices(std::vector<std::vector<double>>& matBlock, std::mt19937_64& PRNG, std::uniform_real_distribution<double>& rngdist, const blasint randomMatSize, const uint32_t numConcurrentThreads, const uint32_t numMat){
|
|
for(uint32_t i=0; i<numMat; i++){
|
|
for(uint32_t j = 0; j < static_cast<uint32_t>(randomMatSize*randomMatSize); j++){
|
|
diff --git a/cpp_thread_test/dgemm_thread_safety.cpp b/cpp_thread_test/dgemm_thread_safety.cpp
|
|
index 1c52875..104c64f 100644
|
|
--- a/cpp_thread_test/dgemm_thread_safety.cpp
|
|
+++ b/cpp_thread_test/dgemm_thread_safety.cpp
|
|
@@ -46,6 +46,8 @@ int main(int argc, char* argv[]){
|
|
std::cout<<"Number of concurrent calls into OpenBLAS : "<<numConcurrentThreads<<'\n';
|
|
std::cout<<"Number of testing rounds : "<<numTestRounds<<'\n';
|
|
std::cout<<"This test will need "<<(static_cast<uint64_t>(randomMatSize*randomMatSize)*numConcurrentThreads*3*8)/static_cast<double>(1024*1024)<<" MiB of RAM\n"<<std::endl;
|
|
+
|
|
+ FailIfThreadsAreZero(numConcurrentThreads);
|
|
|
|
std::cout<<"Initializing random number generator..."<<std::flush;
|
|
std::mt19937_64 PRNG = InitPRNG();
|
|
diff --git a/cpp_thread_test/dgemv_thread_safety.cpp b/cpp_thread_test/dgemv_thread_safety.cpp
|
|
index 5411fec..20ea381 100644
|
|
--- a/cpp_thread_test/dgemv_thread_safety.cpp
|
|
+++ b/cpp_thread_test/dgemv_thread_safety.cpp
|
|
@@ -18,7 +18,7 @@ int main(int argc, char* argv[]){
|
|
uint32_t maxHwThreads = omp_get_max_threads();
|
|
|
|
if (maxHwThreads < 52)
|
|
- numConcurrentThreads = maxHwThreads -4;
|
|
+ numConcurrentThreads = maxHwThreads;
|
|
|
|
if (argc > 4){
|
|
std::cout<<"ERROR: too many arguments for thread safety tester"<<std::endl;
|
|
@@ -47,6 +47,8 @@ int main(int argc, char* argv[]){
|
|
std::cout<<"Number of concurrent calls into OpenBLAS : "<<numConcurrentThreads<<'\n';
|
|
std::cout<<"Number of testing rounds : "<<numTestRounds<<'\n';
|
|
std::cout<<"This test will need "<<((static_cast<uint64_t>(randomMatSize*randomMatSize)*numConcurrentThreads*8)+(static_cast<uint64_t>(randomMatSize)*numConcurrentThreads*8*2))/static_cast<double>(1024*1024)<<" MiB of RAM\n"<<std::endl;
|
|
+
|
|
+ FailIfThreadsAreZero(numConcurrentThreads);
|
|
|
|
std::cout<<"Initializing random number generator..."<<std::flush;
|
|
std::mt19937_64 PRNG = InitPRNG();
|