GitLab

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

Theoretically this might have lower numerical error since the scaling is in
fp32 instead of fp16 (not sure, I haven't thought too carefully about it).
However, in practice, the numerical errors seem about the same.