checkpoint pr be moved here, test=develop

6 years ago · 09442fb27e
parent 622fe6a56b
commit 09442fb27e
4 changed files with 4 additions and 0 deletions
--- 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>);
--- 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>);
--- 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>);
--- 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>);