kho: extend scratch

From: “Pratyush Yadav (Google)” pratyush@kernel.org Motivation ========== The scratch space is allocated by the first kernel in the KHO chain, and is reused by all subsequent kernels. The size of