Skip to content

portable: accumulate in fp32 for Half/BFloat16 in softmax, log_softmax, mean, and sum#20089

Closed
vacu9708 wants to merge 2 commits into
pytorch:mainfrom
vacu9708:accumulate_in_fp32_in_softmax
Closed

portable: accumulate in fp32 for Half/BFloat16 in softmax, log_softmax, mean, and sum#20089
vacu9708 wants to merge 2 commits into
pytorch:mainfrom
vacu9708:accumulate_in_fp32_in_softmax