jax
b1729840 - Lower complex128 scatter-add to two float64 scatter-adds on GPU.

Commit

5 years ago

Lower complex128 scatter-add to two float64 scatter-adds on GPU. This works around a slow scatter for 128-bit wide values.

References

Author

hawkinsp

Parents