enh: support runtime dim for count

[Pranavchiku]

Fixes #3630.

[Pranavchiku]

integration_tests/intrinsic_172.f90 work well for me ( Mac M2 air ), will have to check why it fails on linux.

[Pranavchiku]

This works fine on linux as well:

~/lfortran (gh3630)$ lfortran ./integration_tests/intrinsics_172.f90 
7
6
18
2 1 2 2 
2 2 3 
2 1 2 2 
2 2 3 
6
9 
6
6 
6
6
18
4 3 
18
3 3 
18
18
18
18
57
5 3 1 

~/lfortran (gh3630)$ lfortran ./integration_tests/intrinsics_172.f90  --fast
7
6
18
2 1 2 2 
2 2 3 
2 1 2 2 
2 2 3 
6
9 
6
6 
6
6
18
4 3 
18
3 3 
18
18
18
18
57
5 3 1 

[Thirumalai-Shaktivel]

The tests fail for the LLVM version >= 14. Try creating a new conda env and install LLVM 14. I think you will be able to reproduce the seg fault.

[Pranavchiku]

I tried it all, it works somehow, let me see if I can reproduce.

[certik]

This line seems to segfault for LLVM >= 14.

[certik]

To Debug:

• uncomment all changes in the test, run CI, ensure everything passes. If not, the bug is elsewhere in the PR.

Assuming it is coming from the line 55 in the test, then we need to debug it why it fails. Try to remove it from print, as well as do other simplifications.

On macOS you can also use leaks to check the generated binary and see if discovers any issues.