@triton.jit
def triton_mse_loss_fp16_kernel(output_ptr, target_ptr, out_ptr,
    stride_a_output, stride_a_target, a, b, BLOCK_SIZE_B: tl.constexpr):
    pid = tl.program_id(0)
    offsets_b = tl.arange(0, BLOCK_SIZE_B)
    output = tl.load(output_ptr + pid * stride_a_output + offsets_b, mask=
        offsets_b < b)
    target = tl.load(target_ptr + pid * stride_a_target + offsets_b, mask=
        offsets_b < b)
    output = output.to(tl.float32)
    target = target.to(tl.float32)
    mse = tl.sum((output - target) * (output - target)) / b
    tl.store(out_ptr + pid, mse)
