[sharktank] Extend to test with Python 3.12

[marbre]

Progress on #357

Extends to test with Python 3.12 and newer torch versions on Linux.

For torch.compile support on Python 3.12, torch ge 2.4.0 is required. Testing with torch 2.3.0 is therefore skipped when running Python 3.12. To limit the number of jobs, tests are currently only expanded for Linux, whereas Windows (and unpinning NumPy) should be handled in a follow up.

The error logged for testExportNondecomposed when failing with torch 2.5.1 slows down the CI by more than 20 minutes, therefore the test is now skipped instead of marked xfail.

[ScottTodd]

Nice! I made some small edits to the PR description and left some suggestions for abbreviating the job names

[ScottTodd]

Maybe rename to just "Unit Tests" or "Test" so more of the matrix shows up in the UI

Could also qualify "torch {{ matrix.torch-version }}"

Here's a mock-up to show where the text gets cut off:

[marbre]

I renamed to "Unit Tests" for now but we probably want to harmonize this more across different workflows in the future.

[ScottTodd]

Nice. Definitely better now. Could still include "torch" next to the torch version but it might be obvious enough to some people.

[marbre]

Renamed to "Unit Tests :: ${{ matrix.os }}, ${{ matrix.python-version }}, ${{ matrix.torch-version }}" as for Windows it was (windows-2022, 3.11, 2.3.0) and now have the same order for all jobs. WDYT?

[ScottTodd]

Ah, the ordering helps too. I think having two versions right next to each other can be confusing unless you are very familiar with Python and PyTorch versions. I personally have to unpack 3.11, 2.4.1 and think for a second, compared to 3.11, torch 2.4.1. We don't have many characters to spare in the UI though.

[marbre]

Save another character but we should aim to harmonize the naming in a follow up.