pytorch
002ce5c1 - port addmm to structure kernel (#57417)

Commit
4 years ago
port addmm to structure kernel (#57417) Summary: Port addmm to structure kernel Follow ups - migrate `mm` and `addbmm` to structure - move TORCH_CHECKS currently in `addmm_cpu_impl_` and `addmm_out_cuda_impl` to meta Pull Request resolved: https://github.com/pytorch/pytorch/pull/57417 Reviewed By: bdhirsh Differential Revision: D28291001 Pulled By: walterddr fbshipit-source-id: 4eafaa30a465e225fbb4d2a69a36f1e037df9122
Author
Rong Rong (AI Infra)
Parents
Loading