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.