fix layernorm

2022-12-18 20:37:55 +08:00 · 2022-12-18 20:37:55 +08:00 · 9d516fa68f
parent e66a18a0bf
commit 9d516fa68f
1 changed files with 1 additions and 1 deletions
--- a/chunk_codegen.py
+++ b/chunk_codegen.py
@ -574,7 +574,7 @@ class IndexTracer(object):
            node_idx (int)
        """
        self._assign_index_as_input(node, idx)
-        self._mark_computation(node, idx, [-1, -2])
+        self._mark_computation(node, idx, [-1])

    def _assign_elementwise_index(self, node, idx):
        """