From 09442fb27e680857d343dbc6087870cac468fed5 Mon Sep 17 00:00:00 2001
From: lujun <lujun315023@126.com>
Date: Thu, 21 Mar 2019 02:19:48 +0800
Subject: [PATCH] checkpoint pr be moved here, test=develop

---
 paddle/fluid/operators/load_combine_op.cu | 1 +
 paddle/fluid/operators/load_op.cu         | 1 +
 paddle/fluid/operators/save_combine_op.cu | 1 +
 paddle/fluid/operators/save_op.cu         | 1 +
 4 files changed, 4 insertions(+)

diff --git a/paddle/fluid/operators/load_combine_op.cu b/paddle/fluid/operators/load_combine_op.cu
index f23f4ea5aa..2a42c0daa7 100644
--- a/paddle/fluid/operators/load_combine_op.cu
+++ b/paddle/fluid/operators/load_combine_op.cu
@@ -21,4 +21,5 @@ REGISTER_OP_CUDA_KERNEL(
     ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, float>,
     ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, double>,
     ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int>,
+    ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
     ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int64_t>);
diff --git a/paddle/fluid/operators/load_op.cu b/paddle/fluid/operators/load_op.cu
index dcbef0b592..90f78110f8 100644
--- a/paddle/fluid/operators/load_op.cu
+++ b/paddle/fluid/operators/load_op.cu
@@ -20,4 +20,5 @@ REGISTER_OP_CUDA_KERNEL(
     load, ops::LoadOpKernel<paddle::platform::CUDADeviceContext, float>,
     ops::LoadOpKernel<paddle::platform::CUDADeviceContext, double>,
     ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int>,
+    ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
     ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int64_t>);
diff --git a/paddle/fluid/operators/save_combine_op.cu b/paddle/fluid/operators/save_combine_op.cu
index 71476fd802..bc4478b51b 100644
--- a/paddle/fluid/operators/save_combine_op.cu
+++ b/paddle/fluid/operators/save_combine_op.cu
@@ -21,4 +21,5 @@ REGISTER_OP_CUDA_KERNEL(
     ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, float>,
     ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, double>,
     ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int>,
+    ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
     ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int64_t>);
diff --git a/paddle/fluid/operators/save_op.cu b/paddle/fluid/operators/save_op.cu
index ebb7e1832c..0a778a694e 100644
--- a/paddle/fluid/operators/save_op.cu
+++ b/paddle/fluid/operators/save_op.cu
@@ -21,6 +21,7 @@ REGISTER_OP_CUDA_KERNEL(
     save, ops::SaveOpKernel<paddle::platform::CUDADeviceContext, float>,
     ops::SaveOpKernel<paddle::platform::CUDADeviceContext, double>,
     ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int>,
+    ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
     ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int64_t>,
     ops::SaveOpKernel<paddle::platform::CUDADeviceContext,
                       paddle::platform::float16>);
-- 
GitLab