From d395757fa790f52c84c0948638ba8bf64d274ea1 Mon Sep 17 00:00:00 2001
From: Bryan Massoth <bmassoth@google.com>
Date: Mon, 3 Nov 2025 18:51:07 -0800
Subject: [PATCH] Integrate SparseCore offloading metadata into grouping logic
 to accurately group SparseCore execution with TensorCore.

PiperOrigin-RevId: 827734931
---
 tsl/profiler/lib/context_types.cc | 2 ++
 tsl/profiler/lib/context_types.h  | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/tsl/profiler/lib/context_types.cc b/tsl/profiler/lib/context_types.cc
index 13ebdb901..58e63b17d 100644
--- 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
index 1e07a0382..42389c1db 100644
--- 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.