OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[Kineto x Insight] Add device to activity type map in pytorch (#154253)

Browse Source 
Summary: Update the device to ActivityType Map in pytorch. Need to be exported to github

Test Plan:
Run the ondemand e2e test and insight profiler is triggered during profiling
P1819539581: https://www.internalfb.com/intern/paste/P1819539581/
{F1978519960}

Insight profiler is not enabled when mtia_insight not specifying in config
{F1978527200}

Reviewed By: fenypatel99

Differential Revision: D75246621

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154253
Approved by: https://github.com/Skylion007
This commit is contained in:
Zizeng Meng 2025-05-28 20:36:16 +00:00 committed by PyTorch MergeBot
parent e25074d462
commit f6dcc45c44
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/csrc/profiler/kineto_shim.cpp
View File

@ -393,7 +393,8 @@ c10::DeviceType deviceTypeFromActivity(libkineto::ActivityType activity_type) {
}
// TODO: T151322015
case libkineto::ActivityType::MTIA_CCP_EVENTS:
case libkineto::ActivityType::MTIA_WORKLOADD: {
case libkineto::ActivityType::MTIA_WORKLOADD:
case libkineto::ActivityType::MTIA_INSIGHT: {
// PrivateUse1 kineto backend reuse above ActivityTypes,
// If PrivateUse1 backend enabled, this should return
// c10::DeviceType::PrivateUse1.