frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Profiler] Add test to make sure FunctionEvents are processed lazily (#134359)

Browse Source 
Summary: Create simple test that checks that FunctionEvent build tree happens lazily by checking that the metrics for it changes before and after call.

Test Plan: Make sure test passes in CI

Reviewed By: briancoutinho

Differential Revision: D61685429

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134359
Approved by: https://github.com/briancoutinho
This commit is contained in:
Shivam Raikundalia
2024-08-24 23:03:19 +00:00
committed by PyTorch MergeBot
parent af4c87953e
commit d33d68e326
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/profiler/test_profiler.py
View File

@ -1974,6 +1974,21 @@ assert KinetoStepTracker.current_step() == initial_step + 2 * niters
torch.add(x, y)
self.assertTrue(len(p2.events()) == 0)
@skipIfTorchDynamo("profiler gets ignored if dynamo activated")
def test_lazy_build_tree(self):
with profile() as p:
self.payload()
stats = p._stats()
# Test that the tree is not built
self.assertEqual(stats.function_events_build_tree_call_duration_us, 0)
self.assertEqual(stats.number_of_events, 0)
# Test that the tree is built on demand
p.events()
self.assertGreater(stats.function_events_build_tree_call_duration_us, 0)
self.assertGreater(stats.number_of_events, 0)
class SimpleNet(nn.Module):
def __init__(self) -> None: