Update

jansel · jansel · commit bec1b2894667 · 2024-11-22T19:51:38.000-08:00
[ghstack-poisoned]
diff --git a/test/inductor/test_debug_trace.py b/test/inductor/test_debug_trace.py
@@ -66,9 +66,9 @@ def fn(a, b):
             open(filename / "ir_pre_fusion.txt").read().rstrip(),
             """\
 op0: SchedulerNode(ComputedBuffer)
-op0.writes = [MemoryDep('buf0', c0, {c0: 256}, None)]
+op0.writes = [MemoryDep('buf0', c0, {c0: 256})]
 op0.unmet_dependencies = []
-op0.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256}, None)]
+op0.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256})]
 op0.outputs = [
     buf0: ComputedBuffer
     buf0.layout = FixedLayout('cpu', torch.float32, size=[16, 16], stride=[16, 1])
@@ -93,8 +93,8 @@ def body(self, ops):
 
 
 op1: SchedulerNode(ComputedBuffer)
-op1.writes = [MemoryDep('buf1', c0, {c0: 256}, None)]
-op1.unmet_dependencies = [MemoryDep('buf0', c0, {c0: 256}, None)]
+op1.writes = [MemoryDep('buf1', c0, {c0: 256})]
+op1.unmet_dependencies = [MemoryDep('buf0', c0, {c0: 256})]
 op1.met_dependencies = []
 op1.outputs = [
     buf1: ComputedBuffer
@@ -134,9 +134,9 @@ def body(self, ops):
             open(filename / "ir_post_fusion.txt").read().rstrip(),
             """\
 op0_op1: FusedSchedulerNode(SchedulerNode,SchedulerNode)
-op0_op1.writes = [MemoryDep('buf0', c0, {c0: 256}, None), MemoryDep('buf1', c0, {c0: 256}, None)]
+op0_op1.writes = [MemoryDep('buf0', c0, {c0: 256}), MemoryDep('buf1', c0, {c0: 256})]
 op0_op1.unmet_dependencies = []
-op0_op1.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256}, None)]
+op0_op1.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256})]
 op0_op1.outputs = [
     buf0: ComputedBuffer
     buf0.layout = FixedLayout('cpu', torch.float32, size=[16, 16], stride=[16, 1])
@@ -147,9 +147,9 @@ def body(self, ops):
 ]
 op0_op1.snodes[0] =
 op0: SchedulerNode(ComputedBuffer)
-op0.writes = [MemoryDep('buf0', c0, {c0: 256}, None)]
+op0.writes = [MemoryDep('buf0', c0, {c0: 256})]
 op0.unmet_dependencies = []
-op0.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256}, None)]
+op0.met_dependencies = [MemoryDep('arg0_1', c0, {c0: 256})]
 op0.outputs = [
     buf0: ComputedBuffer
     buf0.layout = FixedLayout('cpu', torch.float32, size=[16, 16], stride=[16, 1])
@@ -173,8 +173,8 @@ def body(self, ops):
         return store
 op0_op1.snodes[1] =
 op1: SchedulerNode(ComputedBuffer)
-op1.writes = [MemoryDep('buf1', c0, {c0: 256}, None)]
-op1.unmet_dependencies = [MemoryDep('buf0', c0, {c0: 256}, None)]
+op1.writes = [MemoryDep('buf1', c0, {c0: 256})]
+op1.unmet_dependencies = [MemoryDep('buf0', c0, {c0: 256})]
 op1.met_dependencies = []
 op1.outputs = [
     buf1: ComputedBuffer
