misc: Replace dimension check in pull_dims #6167

Job	Run time
pytest-ubuntu-py311-gcc11-noomp-base	28m 35s
pytest-ubuntu-py311-gcc11-noomp-adjoint	27m 58s
pytest-ubuntu-py38-gcc12-omp-base	28m 25s
pytest-ubuntu-py38-gcc12-omp-adjoint	27m 4s
pytest-ubuntu-py37-gcc7-omp-base	29m 41s
pytest-ubuntu-py37-gcc7-omp-adjoint	28m 48s
pytest-ubuntu-py310-gcc10-noomp-base	27m 33s
pytest-ubuntu-py310-gcc10-noomp-adjoint	26m 54s
pytest-ubuntu-py312-gcc13-omp-base	41m 5s
pytest-ubuntu-py312-gcc13-omp-adjoint	37m 10s
pytest-ubuntu-py39-gcc9-omp-base	27m 40s
pytest-ubuntu-py39-gcc9-omp-adjoint	26m 44s
pytest-osx-py37-clang-omp-base	4m 30s
pytest-docker-py39-gcc-omp-base	29m 46s
pytest-docker-py39-gcc-omp-adjoint	25m 40s
pytest-docker-py39-icx-omp-base	24m 29s
pytest-docker-py39-icx-omp-adjoint	26m 26s
	7h 48m 28s

Provide feedback