[XLA] Fix undefined behavior on int32 overflow in prng_test

This is calling Square<int32>(65280), the result doesn't fit into an int32.
LLVM takes advantage of this and changes the behavior of the code, making the
test fail when optimization is enabled (e.g. in msan and tsan mode).
Change: 150672767