Skip to content

Commit 5eaaf35

Browse files
crcrparholly1238
andauthored
fix typo: forward->backward (#872)
Co-authored-by: holly1238 <77758406+holly1238@users.noreply.github.com>
1 parent f119819 commit 5eaaf35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

advanced_source/cpp_extension.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1101,7 +1101,7 @@ on it:
11011101
const int threads = 1024;
11021102
const dim3 blocks((state_size + threads - 1) / threads, batch_size);
11031103
1104-
AT_DISPATCH_FLOATING_TYPES(X.type(), "lltm_forward_cuda", ([&] {
1104+
AT_DISPATCH_FLOATING_TYPES(X.type(), "lltm_backward_cuda", ([&] {
11051105
lltm_cuda_backward_kernel<scalar_t><<<blocks, threads>>>(
11061106
d_old_cell.packed_accessor32<scalar_t,2,torch::RestrictPtrTraits>(),
11071107
d_gates.packed_accessor32<scalar_t,3,torch::RestrictPtrTraits>(),

0 commit comments

Comments
 (0)