Reset dynamo state in test_memory_compile_regions

[amd-maradosa]

ROCm/TheRock#6173

Motivation

test_memory_compile_regions in test/test_cuda.py asserts on Dynamo compile-region counts. When the test runs in the same process as other compile tests (e.g. a full test_cuda.py run), compiled artifacts cached by earlier tests leak into this test, making the asserted counts non-deterministic and the test flaky. This PR isolates the test from prior Dynamo state.

CI fail

Technical Details

Added torch._dynamo.reset() at the start of test_memory_compile_regions, before torch.cuda.memory.empty_cache(), so the test starts from a clean Dynamo cache regardless of what ran before it.

Test Plan

Reproduced locally with a small prelude test that runs torch.compile(...) on CUDA before the affected test, in the same process:

python -m pytest .py test_cuda.py -k "<prelude_test> or test_memory_compile_regions" -v

Test Result

Before fix:
- standalone: PASSED
- with a compiling test first: FAILED

After fix:
- standalone: PASSED
- with a compiling test first: PASSED

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[rocm-repo-management-api]

Jenkins build for ad54c5992eda08520997db8ec488eb78d7542809 commit finished as FAILURE
Links: Pipeline Overview / Build artifacts / Test Results