google-deepmind/alphatensor

Implementation

kXborg opened this issue · 1 comments

kXborg commented

How to use the algorithms to multiply two matrices?

matejbalog commented

You can see the testGPUMatrixMultiplicationPrecision test in benchmarking/test_correctness.py, which constructs a Callable algorithm from a factorization, and uses it to multiply two matrices.

  • 👍1