Skip to content

GitLab

Commit 9d516fa6 authored by oahzxl's avatar oahzxl
Browse files

fix layernorm

parent e66a18a0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
chunk_codegen.py
View file @ 9d516fa6
...@@ -574,7 +574,7 @@ class IndexTracer(object): ...@@ -574,7 +574,7 @@ class IndexTracer(object):
node_idx (int) node_idx (int)
""" """
self._assign_index_as_input(node, idx) 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): def _assign_elementwise_index(self, node, idx):
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment