HERA-Team / matvis

Fast matrix-based visibility simulator
MIT License
2 stars 3 forks source link

updates to incorporate the matrix chunking method #79

Open ccain002 opened 9 months ago

ccain002 commented 9 months ago

This update adds the capability to use the matrix chunking method to do matrix multiplication, alongside the full matrix product and vector-vector method. This mode takes a new argument "matsets", which is a list of tuples of arrays of rows and columns, which specify sub-matrices. After the sub-matrices are multiplied, the unique pairs are selected from the results (as specified by antpairs) and the output is only the unique pairs, just like in the vector-vector case. I've tested that the new matrix multiplication method itself works properly with the new inputs, but I haven't checked that the updates to the API are correct yet.