pltpu.SMEM((1024,), f32)，4 KiB：数值检查通过
pltpu.SMEM((131072,), f32)，512 KiB：数值检查通过
pltpu.SMEM((262144,), f32)，1024 KiB：编译失败：RESOURCE_EXHAUSTED: XLA:TPU compile permanent error. Ran out of memory in memory space smem. Used 1.00M of 1.00M smem. Exceeded smem capacity by 136B.