viable/strict/1791212143: [Test][XPU] Refactor `test_flop_counter.py` and enable on XPU (#197249)
- PyTorch: 2311 events in the last 90 days
- PyTorch: 2281th Release in the last 90 days
- Previous: earlier the same day · trunk/91a0edaa2dabeabc8883d722c6055d772ba48a71
What happened
Move device-agnostic tests to TestFlopCounterDevice Enable TestFlopCounterDevice for XPU Move test_nested_attn_backward_flops_with_unequal_qk_value_dims to TestFlopCounter (uses only the meta device) Move SDPA flop tests from TestFlopCounterCUDA to TestFlopCounterDevice and skip them on XPU until the underlying gaps are fixed Drop the CUDA-specific wording from the skipIf reasons of the moved tests Enable test_flex_…
Summary assembled by rule from the sources below