Use fp32 accumulation in SkipLayerNorm/EmbedLayerNorm CUDA kernels #28682
use float as compute type in layer norm
8ef5ab44
add profiling script
fb33e823
address review feedback for profiling scripts
01336e2a
fallback if no nvtx
5549a98c
tianleiwu
merged
26871ab4
into main 60 days ago
tianleiwu
deleted the tlwu/sln_fp32_compute_type branch 60 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub