神刀安全网

Multithreaded matrix multiplication in Rust – Part I

This will be a fairly short post. I’m labeling this Part I as I’m hoping to give a more full report once I see this work through. It would be good to get some feedback and more eyes on this in the mean time though! Lots of credit to bluss here – he provided an awesome library and lots of support (and code to steal) for this work.

Recently I implemented blussmatrixmultiply in rusty-machine . Matrixmultiply is an awesome, lightweight library that provides fast, native implementations of matrix multiplication in Rust.

The gains were pretty huge (up to 10x improvements on my naive implementation) and this inspired me to see if we could get even better by using multithreading.

Parallelizing matrix multiplication isn’t easy and there are a few different approaches. For now I have been playing around with a pretty simple divide and conquer implementation. The idea is that we find the largest dimension and if it is greater than some threshold we split the matrix along it and repeat recursively on the two new halves. Once we are below the threshold we perform our usual matrix multiplication. The algorithm is summarized nicely here . I’m using rayon to achieve the parallelism.

Initial results are looking pretty good!

test linalg::matrix::mat_mul_128_100        ... bench:     213,072 ns/iter (+/- 47,042) test linalg::matrix::mat_paramul_128_100    ... bench:     270,320 ns/ iter (+/- 167,244) test linalg::matrix::mat_mul_128_1000       ... bench:   1,996,190 ns/iter (+/- 66,789) test linalg::matrix::mat_paramul_128_1000   ... bench:   1,253,315 ns/iter (+/- 1,715,936) test linalg::matrix::mat_mul_128_10000      ... bench:  21,215,542 ns/iter (+/- 793,933) test linalg::matrix::mat_paramul_128_10000  ... bench:  11,659,379 ns/iter (+/- 2,533,431) test linalg::matrix::mat_mul_128_100000     ... bench: 211,979,455 ns/iter (+/- 13,750,261) test linalg::matrix::mat_paramul_128_100000 ... bench: 112,330,287 ns/iter (+/- 4,083,264) 

For comparison, here is the old implementation:

test linalg::matrix::mat_mul_128_100        ... bench:   2,078,298 ns/iter (+/- 132,209) test linalg::matrix::mat_mul_128_1000       ... bench:  20,901,834 ns/iter (+/- 576,653) test linalg::matrix::mat_mul_128_10000      ... bench: 228,113,515 ns/iter (+/- 2,512,868) test linalg::matrix::mat_mul_128_100000     ... bench: too damn long /iter (+/- ____) 

If requested I’ll try to add some benchmarks for parallel BLAS.

This was run on my pretty average laptop with 4 cores. The implementation is still basic and so there is some overhead that can be removed quite easily.

What next?

First I’m going to try trimming off some of this overhead. In the process I’ll hopefully make some of the code a little safer to use. I’ve had to break a couple rules to get this implementation up and running (allowing cloning of mutable raw pointers, and Sync / Send ing them too…).

I’ll also take a deeper look at the matrixmultiply library to see if we can introduce the BLIS multithreading – this may give us better performance.

After this I’m going to see if I can optimize some of the other native linear algebra within rusty-machine . Starting with inverses (probably). This won’t be included as part of this series of posts.

转载本站任何文章请注明:转载至神刀安全网,谢谢神刀安全网 » Multithreaded matrix multiplication in Rust – Part I

分享到:更多 ()

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址
分享按钮