File: cpp_thread_safety_common.h

package info (click to toggle)
openblas 0.3.29%2Bds-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 62,984 kB
  • sloc: asm: 1,264,442; ansic: 412,266; fortran: 74,453; makefile: 13,665; sh: 4,892; perl: 4,468; python: 1,555; cpp: 244
file content (63 lines) | stat: -rw-r--r-- 2,498 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
inline void pauser(){
    /// a portable way to pause a program
    std::string dummy;
    std::cout << "Press enter to continue...";
    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++){
			matBlock[i][j] = rngdist(PRNG);
		}
	}
	for(uint32_t i=numMat; i<(numConcurrentThreads*numMat); i+=numMat){
		for(uint32_t j=0; j<numMat; j++){
			matBlock[i+j] = matBlock[j];
		}
	}
}

void FillVectors(std::vector<std::vector<double>>& vecBlock, std::mt19937_64& PRNG, std::uniform_real_distribution<double>& rngdist, const blasint randomMatSize, const uint32_t numConcurrentThreads, const uint32_t numVec){
	for(uint32_t i=0; i<numVec; i++){
		for(uint32_t j = 0; j < static_cast<uint32_t>(randomMatSize); j++){
			vecBlock[i][j] = rngdist(PRNG);
		}
	}
	for(uint32_t i=numVec; i<(numConcurrentThreads*numVec); i+=numVec){
		for(uint32_t j=0; j<numVec; j++){
			vecBlock[i+j] = vecBlock[j];
		}
	}
}

std::mt19937_64 InitPRNG(){
	std::random_device rd;
	std::mt19937_64 PRNG(rd()); //seed PRNG using /dev/urandom or similar OS provided RNG
	std::uniform_real_distribution<double> rngdist{-1.0, 1.0};
	//make sure the internal state of the PRNG is properly mixed by generating 10M random numbers
	//PRNGs often have unreliable distribution uniformity and other statistical properties before their internal state is sufficiently mixed
	for (uint32_t i=0;i<10000000;i++) rngdist(PRNG);
	return PRNG;
}

void PrintMatrices(const std::vector<std::vector<double>>& matBlock, const blasint randomMatSize, const uint32_t numConcurrentThreads, const uint32_t numMat){
	for (uint32_t i=0;i<numConcurrentThreads*numMat;i++){
		std::cout<<i<<std::endl;
		for (uint32_t j = 0; j < static_cast<uint32_t>(randomMatSize); j++){
			for (uint32_t k = 0; k < static_cast<uint32_t>(randomMatSize); k++){
				std::cout<<matBlock[i][j*randomMatSize + k]<<"  ";
			}
			std::cout<<std::endl;
		}
		std::cout<<std::endl;
	}
}