xla
xla copied to clipboard
GpuInstructionFusion is O(n^2)
See for details: https://github.com/google/jax/issues/18787