Warn if AccumulateGrad stream does not match producer node stream (#1680)

soulitzer · meta-codesync[bot] · commit 422f46664070 · 2025-10-31T14:23:32.000-07:00
Summary: Pull Request resolved: #1680 ghstack-source-id: 59641aa32dc6fd027abf3276017432b693aa71f8 Pull-Request-resolved: pytorch/pytorch#165065 Fixes #ISSUE_NUMBER Opening a new PR for codev cc H-Huang awgu wanchaol fegin fduwjj wz337 wconstab d4l3k pragupta msaroufim dcci X-link: pytorch/pytorch#166136 Reviewed By: ngimel Differential Revision: D85360862 Pulled By: soulitzer fbshipit-source-id: 891ba85bcfdfcd191cb9331a7fc19b5fc4fed6cd
diff --git a/python/monarch/gradient/_gradient_generator.cpp b/python/monarch/gradient/_gradient_generator.cpp
@@ -424,7 +424,8 @@ struct GradientGenerator {
         input_nr,
         check_and_reduce(node->node, input_nr, std::move(t)),
         std::nullopt,
-        std::nullopt);
+        std::nullopt,
+        node->node);
   }
 
   InputBuffer& realInputBuffer(NodeState* state) {

Original file line number	Diff line number	Diff line change
`@@ -424,7 +424,8 @@ struct GradientGenerator {`
`424`	`424`	`input_nr,`
`425`	`425`	`check_and_reduce(node->node, input_nr, std::move(t)),`
`426`	`426`	`std::nullopt,`
`427`		`- std::nullopt);`
	`427`	`+ std::nullopt,`
	`428`	`+ node->node);`
`428`	`429`	`}`
`429`	`430`
`430`	`431`	`InputBuffer& realInputBuffer(NodeState* state) {`