pytorch

6c7410dd - sampled_addmm: BSR support (#101163)

sampled_addmm: BSR support (#101163) This PR implements a `sampled_addmm` kernel that works with a BSR mask. Pull Request resolved: https://github.com/pytorch/pytorch/pull/101163 Approved by: https://github.com/cpuhrsch