Previous Python string replacement didn't match. Now using edit tool. Kernel writes raw FP32 logits with gsa*gsb applied. sqrt(softplus) is done in PyTorch after the kernel returns.
Previous Python string replacement didn't match. Now using edit tool. Kernel writes raw FP32 logits with gsa*gsb applied. sqrt(softplus) is done in PyTorch after the kernel returns.