Fix xla_gpu.atomic_rmw lowering for complex<?> arguments.
Fix xla_gpu.atomic_rmw lowering for complex<?> arguments.

It is not possible to directly bitcast a complex to an int, so we have to go
through an alloca.
