Address remaining comments / minor clean-ups from the reviews at https://github.com/Xilinx/mlir-aie/pull/3599
Address remaining comments / minor clean-ups from the reviews at #3599