From f54a8a7d5534a068a4c0707348b42161a51b298a Mon Sep 17 00:00:00 2001
From: Elliott Slaughter <slaughter@cs.stanford.edu>
Date: Thu, 3 Sep 2026 16:20:45 -0700
Subject: [PATCH] Update Realm task IDs.

---
 .../src/realm-execution/tasks/realm_task_registry.cc          | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/lib/realm-execution/src/realm-execution/tasks/realm_task_registry.cc b/lib/realm-execution/src/realm-execution/tasks/realm_task_registry.cc
index 406a2912a3..a4eea6355f 100644
--- a/lib/realm-execution/src/realm-execution/tasks/realm_task_registry.cc
+++ b/lib/realm-execution/src/realm-execution/tasks/realm_task_registry.cc
@@ -65,6 +65,7 @@ Realm::Event register_all_tasks() {
 
   std::vector<task_id_t> task_ids = {
       // Forward tasks
+      task_id_t::BATCHMATMUL_FWD_TASK_ID,
       task_id_t::BATCHNORM_FWD_TASK_ID,
       task_id_t::BROADCAST_FWD_TASK_ID,
       task_id_t::CAST_FWD_TASK_ID,
@@ -90,8 +91,10 @@ Realm::Event register_all_tasks() {
       task_id_t::SPLIT_FWD_TASK_ID,
       task_id_t::TOPK_FWD_TASK_ID,
       task_id_t::TRANSPOSE_FWD_TASK_ID,
+      task_id_t::UPSAMPLE_FWD_TASK_ID,
 
       // Backward tasks
+      task_id_t::BATCHMATMUL_BWD_TASK_ID,
       task_id_t::BATCHNORM_BWD_TASK_ID,
       task_id_t::BROADCAST_BWD_TASK_ID,
       task_id_t::CAST_BWD_TASK_ID,
@@ -117,6 +120,7 @@ Realm::Event register_all_tasks() {
       task_id_t::SPLIT_BWD_TASK_ID,
       task_id_t::TOPK_BWD_TASK_ID,
       task_id_t::TRANSPOSE_BWD_TASK_ID,
+      task_id_t::UPSAMPLE_BWD_TASK_ID,
 
       // Update tasks
       task_id_t::SGD_UPD_NCCL_TASK_ID,