kernel shared axis optimization is now safe