diff --git a/advanced_source/cpp_extension.rst b/advanced_source/cpp_extension.rst index 4d324e01ccb..e842d98edfb 100644 --- a/advanced_source/cpp_extension.rst +++ b/advanced_source/cpp_extension.rst @@ -1101,7 +1101,7 @@ on it: const int threads = 1024; const dim3 blocks((state_size + threads - 1) / threads, batch_size); - AT_DISPATCH_FLOATING_TYPES(X.type(), "lltm_forward_cuda", ([&] { + AT_DISPATCH_FLOATING_TYPES(X.type(), "lltm_backward_cuda", ([&] { lltm_cuda_backward_kernel<<>>( d_old_cell.packed_accessor32(), d_gates.packed_accessor32(),