Add 'projects/rocprofiler-sdk/' from commit 'bf0fad1d5406fbc51403ba1aa9621a9d4a9bce2b'

git-subtree-dir: projects/rocprofiler-sdk git-subtree-mainline: 50a90550e9 git-subtree-split: bf0fad1d54
2025-07-22 22:52:46 +00:00
parent 50a90550e9 bf0fad1d54
commit ad0fb25ed5
1272 changed files with 230117 additions and 0 deletions
@@ -0,0 +1,149 @@
+###############################################################################
+# MIT License
+#
+# Copyright (c) 2023 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+###############################################################################
+
+import sys
+import os
+
+try:
+    import ctypes
+
+    sqlite3lib = ctypes.CDLL("libsqlite3.so")
+except Exception:
+    pass
+
+from . import libpyrocpd
+from .importer import RocpdImportData
+
+__all__ = [
+    "connect",
+    "execute",
+    "read_agents",
+    "read_nodes",
+    "read_processes",
+    "read_threads",
+    "write_perfetto",
+    "write_csv",
+    "write_otf2",
+    "RocpdImportData",
+]
+
+
+def format_path(path, tag=os.path.basename(sys.executable)):
+    return libpyrocpd.format_path(path, tag)
+
+
+def connect(input, *args, **kwargs):
+    return RocpdImportData(input, *args, **kwargs)
+
+
+def execute(data, *args, **kwargs):
+    return data.execute(*args, **kwargs)
+
+
+def read_agents(data, condition=""):
+    return libpyrocpd.read_agents(data, condition)
+
+
+def read_nodes(data, condition=""):
+    return libpyrocpd.read_nodes(data, condition)
+
+
+def read_processes(data, condition=""):
+    return libpyrocpd.read_processes(data, condition)
+
+
+def read_threads(data, condition=""):
+    return libpyrocpd.read_threads(data, condition)
+
+
+def write_perfetto(connection, config=None, **kwargs):
+    """
+    Write Perfetto pftrace output file
+
+    Args:
+        connection (rocpd.RocpdImportData):
+            rocPD instance of database connection(s)
+        config (rocpd.output_config.output_config):
+            Output specification
+
+    Returns:
+        bool: returns True if successful
+    """
+    from . import output_config
+
+    config = (
+        output_config.output_config(**kwargs)
+        if config is None
+        else config.update(**kwargs)
+    )
+
+    return libpyrocpd.write_perfetto(connection, config)
+
+
+def write_csv(connection, config=None, **kwargs):
+    """
+    Write CSV output file(s)
+
+    Args:
+        connection (rocpd.RocpdImportData):
+            rocPD instance of database connection(s)
+        config (rocpd.output_config.output_config):
+            Output specification
+
+    Returns:
+        bool: returns True if successful
+    """
+    from . import output_config
+
+    config = (
+        output_config.output_config(**kwargs)
+        if config is None
+        else config.update(**kwargs)
+    )
+
+    return libpyrocpd.write_csv(connection, config)
+
+
+def write_otf2(connection, config=None, **kwargs):
+    """
+    Write OTF@ output file
+
+    Args:
+        connection (rocpd.RocpdImportData):
+            rocPD instance of database connection(s)
+        config (rocpd.output_config.output_config):
+            Output specification
+
+    Returns:
+        bool: returns True if successful
+    """
+    from . import output_config
+
+    config = (
+        output_config.output_config(**kwargs)
+        if config is None
+        else config.update(**kwargs)
+    )
+
+    return libpyrocpd.write_otf2(connection, config)