Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
Signed-off-by: Mayank Mishra <[email protected]>
  • Loading branch information
mayank31398 committed Dec 18, 2024
1 parent d74969d commit dd418c9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cute_kernels/cute_inductor/rmsnorm.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
def replace_rmsnorm(gm: GraphModule, node: Node) -> None:
if node.op == CALL_FUNCTION and node.target == torch.rms_norm:
with gm.graph.inserting_after(node):
args = node.args
args = list(node.args)
kwargs = node.kwargs

# delete normalized_shape from the args
Expand Down

0 comments on commit dd418c9

Please sign in to comment.