github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 17:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix layernorm

Browse Source
This commit is contained in:
oahzxl
2022-12-18 20:37:55 +08:00
parent e66a18a0bf
commit 9d516fa68f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chunk_codegen.py
View File

@@ -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):
"""