fix tests

Author: Vikas-kum

tests/profiler/pytorch/test_pytorch_profiler.py

@@ -81,4 +81,4 @@ def test_pytorch_profiler(pytorch_profiler_config_parser, out_dir):
     lt.refresh_event_file_list()
     events = lt.get_events(0, time.time() * 1000000)
     print(f"Number of events {len(events)}")
-    assert len(events) == 379
+    assert len(events) == 447

tests/profiler/pytorch/test_pytorch_profiler_rnn.py

@@ -72,5 +72,5 @@ def test_pytorch_profiler_rnn(pytorch_profiler_config_parser, out_dir):
     lt.refresh_event_file_list()
     events = lt.get_events(0, time.time() * 1000000)
     print(f"Number of events {len(events)}")
-    assert len(events) == 59
+    assert len(events) == 61
     shutil.rmtree(out_dir, ignore_errors=True)