github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-03 10:36:47 +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
Show Stats 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) 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):
""" """