From 94fa9d1a957d3faecb5a15cb3c8d0c0f5c7eabf7 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Tue, 19 Sep 2017 22:33:01 +0800
Subject: [PATCH] Remove const cast for device context

---
 paddle/operators/crop_op.cu | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/paddle/operators/crop_op.cu b/paddle/operators/crop_op.cu
index f499ce3f27..05782145b8 100644
--- a/paddle/operators/crop_op.cu
+++ b/paddle/operators/crop_op.cu
@@ -91,12 +91,11 @@ void CropCUDAFunctoin(const framework::ExecutionContext& context) {
   int block = 512;
   int grid = (n * d + block - 1) / block;
 
-  auto* device_context =
-      const_cast<platform::DeviceContext*>(context.device_context_);
-  CropKernel<T,
-             D><<<grid, block, 0,
-                  reinterpret_cast<platform::CUDADeviceContext*>(device_context)
-                      ->stream()>>>(
+  CropKernel<
+      T,
+      D><<<grid, block, 0, reinterpret_cast<const platform::CUDADeviceContext&>(
+                               context.device_context())
+                               .stream()>>>(
       out_count, out_shape_gpu.data<int64_t>(), x_shape_gpu.data<int64_t>(),
       crop_rules_gpu.data<int>(), x_data, out_data);
 }
-- 
GitLab