[ROCm][TunableOp] TunableOp Context Manager for unit tests (#149930)

Browse Source

This PR is cleanup only. There are no feature changes or bug fixes.

We create a TunableOp context manager for setting up and cleanup. We re-write TunableOp unit tests in terms of this context manager. Ultimately reduces the amount of copy-paste code.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/149930
Approved by: https://github.com/jeffdaily

...

This commit is contained in:

Nichols A. Romero

2025-03-26 02:59:54 +00:00

committed by PyTorch MergeBot

parent a8d0c5c928

commit 45b11730f1

1 changed files with 313 additions and 556 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

869

test/test_linalg.py

File diff suppressed because it is too large Load Diff