Add new SparseCore offloading ContextType.

bmass02 · copybara-github · commit 62db3b3c3b54 · 2025-12-04T21:17:48.000-08:00
PiperOrigin-RevId: 840536042
diff --git a/tsl/profiler/lib/context_types.cc b/tsl/profiler/lib/context_types.cc
@@ -52,6 +52,8 @@ const char* GetContextTypeString(ContextType context_type) {
       return "threadpool_event";
     case ContextType::kJaxServingExecutor:
       return "jax_serving";
+    case ContextType::kScOffload:
+      return "sparsecore_offload";
   }
 }
 
diff --git a/tsl/profiler/lib/context_types.h b/tsl/profiler/lib/context_types.h
@@ -39,7 +39,8 @@ enum class ContextType : int {
   kPjrtLibraryCall,
   kThreadpoolEvent,
   kJaxServingExecutor,
-  kLastContextType = ContextType::kJaxServingExecutor,
+  kScOffload,
+  kLastContextType = ContextType::kScOffload,
 };
 
 // In XFlow we encode context type as flow category as 6 bits.

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,8 @@ const char* GetContextTypeString(ContextType context_type) {`
`52`	`52`	`return "threadpool_event";`
`53`	`53`	`case ContextType::kJaxServingExecutor:`
`54`	`54`	`return "jax_serving";`
	`55`	`+ case ContextType::kScOffload:`
	`56`	`+ return "sparsecore_offload";`
`55`	`57`	`}`
`56`	`58`	`}`
`57`	`59`