#include<iostream>
#include"kronlib.h"usingnamespacekronlib;intmain() {
HostMatrix<float> mat1(2,2);
mat1 = 1, 2,
3, 4;
auto ans1 = mat1 * mat1;
auto ans2 = mat1.elemwise_mult(ans1);
auto ans3 = mat1.dot(Trans, mat1, None);
HostMatrix<double> dmat = ans3;
Cholesky<HostMatrix<float>> chol{ dmat };
std::cout << chol.solve(dmat);
Kronecker<HostMatrix<float>> kron{ {ans1, ans2, ans3} };
CUDAMatrix<float> cuda{ mat1 };
CUDAMatrix<float> one(100, 100);
CUDAMatrix<float> two(500, 100);
CUDAMatrix<float> three(100, 100);
KroneckerVectorStack<CUDAMatrix<float>> kvs{ {one, two, three} };
CUDAMatrix<float> leftvec(1, 100 * 500 * 100);
std::cout << dot(leftvec, kvs).nR() << std::endl;
std::cout << dot(leftvec, kvs).nC() << std::endl;
return0;
}Latest commit
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|