Skip to content

Commit

Permalink
Update example notebook to use do_loop_fusion
Browse files Browse the repository at this point in the history
  • Loading branch information
awnawab committed Nov 8, 2024
1 parent 1437de6 commit f3032e2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions example/03_loop_fusion.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -923,7 +923,7 @@
},
{
"cell_type": "code",
"execution_count": 23,
"execution_count": null,
"id": "164b5054",
"metadata": {},
"outputs": [
Expand Down Expand Up @@ -982,8 +982,8 @@
}
],
"source": [
"from loki import loop_fusion\n",
"loop_fusion(routine_copy)\n",
"from loki import do_loop_fusion\n",
"do_loop_fusion(routine_copy)\n",
"pragma_map = {pragma: None for pragma in FindNodes(Pragma).visit(routine_copy.body)}\n",
"routine_copy.body = Transformer(pragma_map).visit(routine_copy.body)\n",
"print(routine_copy.to_fortran())"
Expand Down

0 comments on commit f3032e2

Please sign in to comment.