[rocprof-compute] Adding Triton backend to marker injection

[ggottipa-amd]

Motivation

This PR enables operator tracing for Triton workloads.

Technical Details

1. Wraps Triton's launch entry points — JITFunction.run for eager kernels and CompiledKernel.run/call for compiled/Inductor kernels — to emit a ROCTX marker per launch, attributed to the user's call site.
2. Adds --triton-trace (and --ml-api-trace for all backends) for profiling and --list-triton-operators / --triton-operator for analysis, driven by a shared backend registry.
3. Each marker is tagged with its framework so Triton and Torch kernels are analyzed independently.
4. Marker names are percent-encoded so names containing slashes don't break parsing, with identical encode/decode in the Python tier and the analyzer.
5. Adds Triton backend unit tests (launch-point wrapping, reentrancy, kernel-name resolution, percent-encode round trip), C++ gtests for encode/decode symmetry, framework-selection and per-backend analyze-filter tests, and a GPU+Triton-gated end-to-end profile-and-analyze test over a sample Triton workload.
6. Adds two cached profiling workloads: tests/workloads/triton_trace/MI300A and tests/workloads/ml_api_trace/MI300A, so the per-backend analyze tests run CPU-only in CI.

JIRA ID

AIPROFCOMP-635

Test Plan

Triton backend unit tests (wrapping, reentrancy dedup, kernel-name, percent-encode round trip)
ctest -R test_inject_roctx_package

Framework selection (flag -> backend, env var)
ctest -R test_profiler_base

Analyze: per-backend operator filtering
ctest -R test_analyze_workloads

Utils: operator pattern parsing
ctest -R test_utils

C++ gtests: percent-encode + round-trip decode
ctest -R test-roctx-recordfn

End-to-end Triton profile + analyze (requires GPU + Triton)
ctest -R test_profile_triton_trace

Test Result

Tests Pass

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[Copilot]

Pull request overview

This PR extends rocprofiler-compute’s ROCTX marker injection and analysis pipeline to support a Triton backend alongside the existing PyTorch backend, adding new CLI flags, backend-aware operator filtering/listing, and shared Python-tier ROCTX initialization.

Changes:

• Add Triton tracing/profile CLI support (--triton-trace) and analysis CLI support (--list-triton-operators, --triton-operator) with updated docs.
• Implement and test a hardened Triton ROCTX injection backend that wraps Triton kernel launch entry points and attributes markers to a Backend column.
• Generalize analyze-side operator selection/listing to be backend-aware (torch vs triton), including per-backend matched trace storage.

Reviewed changes

Copilot reviewed 13 out of 13 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
projects/rocprofiler-compute/tests/test_utils.py	Adds tests for backend-aware operator pattern parsing and backend filtering in analysis CLI.
projects/rocprofiler-compute/tests/test_profiler_base.py	Adds tests validating CLI framework selection and subprocess env var construction.
projects/rocprofiler-compute/tests/test_inject_roctx_module.py	Adds Triton backend unit tests (wrapping behavior, reentrancy guard, python-tier handling, kernel name extraction).
projects/rocprofiler-compute/src/utils/tty.py	Generalizes operator listing output to support a configurable framework label (PyTorch/Triton).
projects/rocprofiler-compute/src/utils/schema.py	Extends Workload to store matched API-trace rows per backend.
projects/rocprofiler-compute/src/utils/inject_roctx/_core.py	Introduces shared Python-tier ROCTX initialization (ensure_python_tier) and candidate path tracking.
projects/rocprofiler-compute/src/utils/inject_roctx/_backends/_triton.py	Adds/extends Triton backend wrapping logic for kernel launches and framework-root registration.
projects/rocprofiler-compute/src/utils/inject_roctx/_backends/_torch.py	Switches torch backend to use shared Python-tier initialization and shared candidate-path reporting.
projects/rocprofiler-compute/src/rocprof_compute_profile/profiler_base.py	Adds Triton framework selection mapping for ROCTX injection env var.
projects/rocprofiler-compute/src/rocprof_compute_analyze/analysis_cli.py	Refactors operator list/filter flow to be backend-aware and adds backend filtering on consolidated API traces.
projects/rocprofiler-compute/src/argparser.py	Adds CLI flags for Triton tracing and Triton operator listing/filtering.
projects/rocprofiler-compute/docs/how-to/profile/mode.rst	Documents Triton trace usage, requirements, and relationship to ROCTX frameworks env var.
projects/rocprofiler-compute/docs/how-to/analyze/cli.rst	Documents Triton operator analysis and its CLI usage.

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.