Add trivial gemm test for multithread consistency
This commit is contained in:
parent
62f4c84f27
commit
f5fcc5baec
|
@ -0,0 +1,20 @@
|
|||
#include <iostream>
|
||||
#include <cblas.h>
|
||||
int main ( int argc, char* argv[] ) {
|
||||
const long n = ((long)1 << 31) - 1;
|
||||
std::cout << n <<std::endl;
|
||||
float* A = new float[n];
|
||||
float* B = new float[n];
|
||||
float* C = new float[1];
|
||||
for(long i =0; i <n; i++){
|
||||
A[i] = 1;
|
||||
B[i] = 1;
|
||||
|
||||
}
|
||||
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, 1, 1, n, 1.0, A, n, B, 1, 0.0, C, 1);
|
||||
std::cout << *C <<std::endl;
|
||||
delete[] A;
|
||||
delete[] B;
|
||||
delete[] C;
|
||||
return 0;
|
||||
}
|
Loading…
Reference in New Issue