search

libxsmm / libxsmm

Library for specialized dense and sparse matrix operations, and deep learning primitives.
https://libxsmm.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
846 stars 182 forks source link

Integrate with Pytorch #882

Closed akote123 closed 1 month ago

akote123 commented 5 months ago

I have tried integrating libxsmm with pytorch by LD_PRELOAD but I am unable to see the lib in perf . Is ther any other method I can integrate this library with Pytorch.

alheinecke commented 1 month ago

https://github.com/libxsmm/tpp-pytorch-extension