Skip to content

Commit 82745f4

Browse files
committed
Native op, C++ syntax highlighting
1 parent 25928b3 commit 82745f4

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

returnn/native_op.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1017,6 +1017,7 @@ def grad_input_map(cls, X, W, y0, c0, i, start, step, Y, C, H, d, DY, DC, DH, Dd
10171017
return (X, W, y0, c0, i, start, step, Y, C, H, DY, Dd)
10181018

10191019
c_extra_support_code = {
1020+
# language=C++
10201021
"lstm_kernel": """
10211022
DEF_KERNEL
10221023
void lstm_kernel(
@@ -1058,6 +1059,7 @@ def grad_input_map(cls, X, W, y0, c0, i, start, step, Y, C, H, d, DY, DC, DH, Dd
10581059
}
10591060
}
10601061
""",
1062+
# language=C++
10611063
"lstm_bwd_kernel": """
10621064
DEF_KERNEL
10631065
void lstm_bwd_kernel(
@@ -1118,6 +1120,7 @@ def grad_input_map(cls, X, W, y0, c0, i, start, step, Y, C, H, d, DY, DC, DH, Dd
11181120
""",
11191121
}
11201122

1123+
# language=C++
11211124
c_fw_code = """
11221125
// X, W, y0, c0, i, start, step = input_names
11231126
// Y, C, H, d = output_names

0 commit comments

Comments
 (0)