try {

// Set matrix size

int size = 5;

// Input variables

std::complex<double> i = std::complex<double>(0, 1);

CUDAMatrix A(size, {

1, 0, 0, 0, 0,

0, 2, 0, 0, 0,

0, 0, 3, 0, 0,

0, 0, 0, 4, 0,

0, 0, 0, 0, 5

});

// Result variables

CUDAMatrix eA(size);

CUDAMatrix eAi(size);

// Create timers

CUDATimer t1, t2;

// Calculations

t1 = CUDAMatrix::exp(A, eA);

t2 = CUDAMatrix::mul(eA, i, eAi);

// Output

std::cout << "A" << A << std::endl;

std::cout << "e^A" << eA << t1 << std::endl;

std::cout << "e^A * i" << eAi << t2 << std::endl;

} catch (std::exception e) {

std::cout << std::endl << e.what() << std::endl;

}

return 0;