hevc: Special case square hls_prediction_unit