I’m sorry, I posted my answer as I was typing it, by mistake.
I looked into this further, it looks like a bug that we introduced in 1.0.0, I’m taking a further look and filing an issue.
It is only affected on the CPU, and produces the correct result on GPU.