From 5ee7a72cd6ca14a3f77f402b12fd7fa808f62caa Mon Sep 17 00:00:00 2001
From: Kate Unger <kateunger@ucsb.edu>
Date: Wed, 13 Sep 2023 12:28:20 -0700
Subject: [PATCH] conv2d typo

---
 lib/runtime/src/ops/conv_2d.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/runtime/src/ops/conv_2d.cc b/lib/runtime/src/ops/conv_2d.cc
index e0e1872a06..5acee3813e 100644
--- a/lib/runtime/src/ops/conv_2d.cc
+++ b/lib/runtime/src/ops/conv_2d.cc
@@ -229,7 +229,7 @@ void register_task<CONV2D_FWD_TASK_ID>() {
 
   fwd.add_arg_slot<bool>(PROFILING);
   fwd.add_unchecked_arg_slot<Conv2DPerDeviceState>(PER_DEVICE_STATE);
-  init.add_arg_slot<Conv2DAttrs>(ATTRS);
+  fwd.add_arg_slot<Conv2DAttrs>(ATTRS);
 
   fwd.add_input_slot(INPUT);
   fwd.add_output_slot(OUTPUT);