[PROTON] Enable the default session_id for activate and deactivate interfaces (#3771)

Author: Jokeren

third_party/proton/proton/profile.py

@@ -54,27 +54,27 @@ def start(
     return libproton.start(name, backend, context, data)
 
 
-def activate(session: int) -> None:
+def activate(session: Optional[int] = 0) -> None:
     """
     Activate the specified session.
     The profiling session will be active and data will be recorded.
 
     Args:
-        session (int): The session ID of the profiling session.
+        session (int): The session ID of the profiling session. Defaults to 0 (the first session started.)
 
     Returns:
         None
     """
     libproton.activate(session)
 
 
-def deactivate(session: int) -> None:
+def deactivate(session: Optional[int] = 0) -> None:
     """
     Stop the specified session.
     The profiling session's data will still be in the memory, but no more data will be recorded.
 
     Args:
-        session (int): The session ID of the profiling session.
+        session (int): The session ID of the profiling session. Defaults to 0 (the first session started.)
 
     Returns:
         None

third_party/proton/test/test_api.py

@@ -7,9 +7,10 @@
 def test_profile():
     with tempfile.NamedTemporaryFile(delete=True, suffix=".hatchet") as f:
         session_id0 = proton.start(f.name.split(".")[0])
-        proton.activate(session_id0)
-        proton.deactivate(session_id0)
+        proton.activate()
+        proton.deactivate()
         proton.finalize()
+        assert session_id0 == 0
 
     with tempfile.NamedTemporaryFile(delete=True, suffix=".hatchet") as f:
         session_id1 = proton.start(f.name.split(".")[0])

third_party/proton/test/test_viewer.py

@@ -1,8 +1,7 @@
-import triton.profiler as proton
 import subprocess
 
 
 def test_help():
     # Only check if the viewer can be invoked
-    ret = subprocess.check_call(["proton-viewer", "-h"])
+    ret = subprocess.check_call(["proton-viewer", "-h"], stdout=subprocess.DEVNULL)
     assert ret == 0