Skip to content

aiupti plugin: map SPYRE_CODE_HOST_COMPUTE and backfilled cbids to names - #44

Open
prasannabiswas-iitb wants to merge 1 commit into
IBM:mainfrom
prasannabiswas-iitb:profiler/jobplan-step-logging
Open

aiupti plugin: map SPYRE_CODE_HOST_COMPUTE and backfilled cbids to names#44
prasannabiswas-iitb wants to merge 1 commit into
IBM:mainfrom
prasannabiswas-iitb:profiler/jobplan-step-logging

Conversation

@prasannabiswas-iitb

Copy link
Copy Markdown
Contributor

Summary

Maps the new and backfilled CBIDs (38-41) to human-readable trace event names in runtimeCbidName():

CBID Enum Display Name
38 AIUPTI_RUNTIME_TRACE_CBID_PARSE_RESPONSE aiuParseResponse
39 AIUPTI_RUNTIME_TRACE_CBID_ISSUE_CALLBACK aiuIssueCallback
40 AIUPTI_RUNTIME_TRACE_CBID_VERIFY_ASYC_MSGS aiuVerifyAsyncMsgs
41 AIUPTI_RUNTIME_TRACE_CBID_SPYRE_CODE_HOST_COMPUTE aiuSpyreCodeHostCompute

These names appear in the "name" field of Chrome trace events (category: privateuse1_runtime).

Companion PRs

  • flex: Instrumentation source — ai-chip-toolchain/flex branch profiler/jobplan-step-logging
  • libaiupti: CBID enum sync — ai-chip-toolchain/libaiupti branch profiler/jobplan-step-logging

Overhead Analysis

Minimal runtime overhead. Compared spyre-perf-suite numbers generated by this branch v/s main branch.

Trace Evidence

Confirmed in PyTorch profiler output:

{"name": "aiuSpyreCodeHostCompute", "cat": "privateuse1_runtime", "dur": 256.784}
{"name": "aiuDataConvert", "cat": "privateuse1_runtime", "dur": 150.415}

mean_mul_softmax_trace.json.txt

Test Plan

  • New CBID names appear correctly in exported Chrome trace JSON
  • Existing CBID names unchanged (no regression)
  • Verified with mean_mul_softmax workload on hardware

Signed-off-by: Antoni Viros i Martin <aviros@ibm.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants