Habana Labs interview question

Implement a function that mul two matrix