Batch matrix exponential computation with CPU/GPU

版本 1.0.3 (79.7 KB) 作者: Yi-Hao Chen
Calculate several matrix exponentials at a time with Pade approximation
1.0
(1)
147.0 次下载
更新时间 2018/12/4

查看许可证

This code is based on Ken Johnson's "expm_". I extended it to compute with GPU. Besides, it can compute for several matrices at a time by using Pade approximation. The matrices should be stored in each page of a multidimensional array.
For CPU computation, please install Yuval's "mmx" for a faster matrix computation, otherwise, it uses a for-loop.

This code can be 400x faster than using for-loop Matlab "expm" when calculating 10000 5x5 randn matrices under GPU. It takes only 0.04s, while Matlab "expm" takes 20s.

Future plan: For Hermitian (or possibly anti-Hermitian) matrices, calculate the eigenvectors and eigenvalues for multiple matrices at once with GPU with householder transformation and QR decomposition, and then use them to calculate expm, which should be faster and this is what Matlab expm does when it's a hermitian matrix. This will take a long time since I'm not familiar with these two methods and I can only do it "if I have enough free time".

引用格式

Yi-Hao Chen (2025). Batch matrix exponential computation with CPU/GPU (https://www.mathworks.com/matlabcentral/fileexchange/67668-batch-matrix-exponential-computation-with-cpu-gpu), MATLAB Central File Exchange. 检索时间: .

MATLAB 版本兼容性
创建方式 R2018b
兼容任何版本
平台兼容性
Windows macOS Linux
类别
在 Help Center 和 MATLAB Answers 中查找有关 Linear Algebra 的更多信息
致谢

参考作品: mmx, Matrix exponential, Matrix polynomial

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

myexpm_

版本 已发布 发行说明
1.0.3

Bug fixed. I compared a bunch of random matrices and it works well. It should be fine now.
1.0.1.2

Still has bugs
1.0.1.1

Bug fixed.
Besides, the previous comment on the bug is wrong. It's that the calculation of the number of terms of Pade approximation is wrong.
1.0.1

Please stop using this. I found a bug if the dimension of the input array is more than 4. That is to say, it can only deal with (n,n,num_matrix) cases
1.0.0.2

Put in the benchmark for the comparison of Matlab expm with for loop and this "myexpm_" function.
1.0.0.1

Fix a bug in multslash
1.0.0.0

Modify user information
Add some user information into the code.