pytorch

54dcb0ea - NT support for matmul of (B, *, C, D) NT with dense (D, E) (#108370)

NT support for matmul of (B, *, C, D) NT with dense (D, E) (#108370) Used in SAM Pull Request resolved: https://github.com/pytorch/pytorch/pull/108370 Approved by: https://github.com/cpuhrsch ghstack dependencies: #108361