jax.experimental.pallas.mosaic_gpu.set_max_registers# jax.experimental.pallas.mosaic_gpu.set_max_registers(n, *, action)[原始碼]# 設定一個 warp 擁有的最大暫存器數量。 參數: n (int) action (Literal['increase', 'decrease'])
