changed first shader stage such that output norm is equal to the later, in-place norm computation in cpu version