torchtitan icon indicating copy to clipboard operation
torchtitan copied to clipboard
verified publisher icon pytorch

register fused rmsnorm as pytorch custom op

Open tianyu-l opened this issue 1 year ago • 0 comments

Stack from ghstack (oldest at bottom):

  • -> #296

tianyu-l avatar May 02 '24 21:05 tianyu-l