From 3480e345700e9750c7f6e76a03955aa75b5299f1 Mon Sep 17 00:00:00 2001
From: lijing-mt <jing.li@mthreads.com>
Date: Thu, 25 Jun 2026 14:53:38 +0800
Subject: [PATCH] update torch_musa 2.9.1

---
 libkineto/src/MuptiActivity.cpp | 15 ++++++++++-----
 libkineto/src/mupti_strings.cpp | 27 ++++++++++++++++++---------
 2 files changed, 28 insertions(+), 14 deletions(-)

diff --git a/libkineto/src/MuptiActivity.cpp b/libkineto/src/MuptiActivity.cpp
index 0ed68c6..9b889da 100644
--- a/libkineto/src/MuptiActivity.cpp
+++ b/libkineto/src/MuptiActivity.cpp
@@ -367,7 +367,9 @@ inline const std::string RuntimeActivity::metadataJson() const {
 }
 
 inline bool isKernelLaunchApi(const MUpti_ActivityAPI& activity_) {
-  return activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernel;
+  return activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernel ||
+      activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernelEx ||
+      activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernelEx_ptsz;
 }
 
 inline bool DriverActivity::flowStart() const {
@@ -381,14 +383,17 @@ inline const std::string DriverActivity::metadataJson() const {
 }
 
 inline const std::string DriverActivity::name() const {
-  // currently only muLaunchKernel is expected
   assert(isKernelLaunchApi(activity_));
-  // not yet implementing full name matching
   if (activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernel) {
     return "muLaunchKernel";
-  } else {
-    return "Unknown"; // should not reach here
   }
+  if (activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernelEx) {
+    return "muLaunchKernelEx";
+  }
+  if (activity_.cbid == MUPTI_DRIVER_TRACE_CBID_muLaunchKernelEx_ptsz) {
+    return "muLaunchKernelEx_ptsz";
+  }
+  return "Unknown"; // should not reach here
 }
 
 template<class T>
diff --git a/libkineto/src/mupti_strings.cpp b/libkineto/src/mupti_strings.cpp
index acbc8b9..15fad5d 100644
--- a/libkineto/src/mupti_strings.cpp
+++ b/libkineto/src/mupti_strings.cpp
@@ -482,7 +482,7 @@ static const char* runtimeCbidNames[] = {
     "musaStreamGetCaptureInfo_ptsz",
     "musaGraphExecKernelNodeSetParams",
     "musaThreadExchangeStreamCaptureMode",
-    "musaDeviceGetNvSciSyncAttributes",
+    "musaDeviceGetMtSciSyncAttributes",
     "musaOccupancyAvailableDynamicSMemPerBlock",
     "musaStreamSetFlags",
     "musaStreamSetFlags_ptsz",
@@ -547,10 +547,10 @@ static const char* runtimeCbidNames[] = {
     "musaMemPoolImportPointer",
     "musaMallocFromPoolAsync",
     "musaMallocFromPoolAsync_ptsz",
-    "musaSignalExternalSemaphoresAsync",
-    "musaSignalExternalSemaphoresAsync",
-    "musaWaitExternalSemaphoresAsync",
-    "musaWaitExternalSemaphoresAsync",
+    "musaSignalExternalSemaphoresAsync_v2",
+    "musaSignalExternalSemaphoresAsync_v2_ptsz",
+    "musaWaitExternalSemaphoresAsync_v2",
+    "musaWaitExternalSemaphoresAsync_v2_ptsz",
     "musaGraphAddExternalSemaphoresSignalNode",
     "musaGraphExternalSemaphoresSignalNodeGetParams",
     "musaGraphExternalSemaphoresSignalNodeSetParams",
@@ -634,7 +634,7 @@ static const char* runtimeCbidNames[] = {
     "musa477",
     "musa478",
     "musa479",
-    "musaStr",
+    "musaStreamGetDevice",
     "musaStreamGetDevice_ptsz",
     "musaMemcpyBatchAsync",
     "musaMemcpyBatchAsync_ptsz",
@@ -660,14 +660,23 @@ static const char* runtimeCbidNames[] = {
     "musaGetExportTable",
     "musaMemoryTransfer",
     "musaMemoryTransfer_ptds",
-    "musaMemoryTransferAsync ",
+    "musaMemoryTransferAsync",
     "musaMemoryTransferAsync_ptsz",
     "musaGraphAddMemTransferNode",
     "musaGraphMemTransferNodeGetParams",
     "musaGraphMemTransferNodeSetParams",
     "musaGraphExecMemTransferNodeSetParams",
-    "SIZE",
-    "FORCE_INT"
+    "musaKernelSetAttributeForDevice",
+    "musaLibraryEnumerateKernels",
+    "musaLibraryGetGlobal",
+    "musaLibraryGetKernel",
+    "musaLibraryGetKernelCount",
+    "musaLibraryGetManaged",
+    "musaLibraryGetUnifiedFunction",
+    "musaLibraryLoadData",
+    "musaLibraryLoadFromFile",
+    "musaLibraryUnload",
+    "SIZE"
 };
 
 const char* runtimeCbidName(MUpti_CallbackId cbid) {