Merge pull request pytorch#358 from tczhangzhi/patch-1

Fix wrong parameter.
qinwentu · Nov 19, 2018 · 400e21b · 400e21b
2 parents 31d9132 + 44d0315
commit 400e21b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/advanced_source/cpp_extension.rst b/advanced_source/cpp_extension.rst
@@ -954,7 +954,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<scalar_t><<<blocks, threads>>>(
           d_old_cell.data<scalar_t>(),
           d_gates.data<scalar_t>(),