From 00360e7eb5c1833f1484a05d425a3938de055475 Mon Sep 17 00:00:00 2001
From: typhoonzero <typhoonzero1986@gmail.com>
Date: Tue, 7 Nov 2017 18:13:28 +0800
Subject: [PATCH] update

---
 paddle/operators/lookup_table_op.cu | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/paddle/operators/lookup_table_op.cu b/paddle/operators/lookup_table_op.cu
index 10d66e5ff4..84b044184a 100644
--- a/paddle/operators/lookup_table_op.cu
+++ b/paddle/operators/lookup_table_op.cu
@@ -74,8 +74,9 @@ class LookupTableCUDAKernel : public framework::OpKernel<T> {
 
     dim3 threads(128, 8);
     dim3 grids(8, 1);
-    LookupTable<T, 128, 8,
-                8><<<grids, threads, 0, context.device_context().stream()>>>(
+    LookupTable<
+        T, 128, 8,
+        8><<<grids, threads, 0, context.cuda_device_context().stream()>>>(
         output, table, ids, N, K, D);
   }
 };
@@ -135,7 +136,7 @@ class LookupTableGradCUDAKernel : public framework::OpKernel<T> {
       dim3 grids(8, 1);
       LookupTableGrad<
           T, 128, 8,
-          8><<<grids, threads, 0, context.device_context().stream()>>>(
+          8><<<grids, threads, 0, context.cuda_device_context().stream()>>>(
           d_table, d_output, ids, N, K, D);
     }
   }
-- 
GitLab