Skip to content

Commit 09442fb

Browse files
committed
checkpoint pr be moved here, test=develop
1 parent 622fe6a commit 09442fb

File tree

4 files changed

+4
-0
lines changed

4 files changed

+4
-0
lines changed

paddle/fluid/operators/load_combine_op.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,5 @@ REGISTER_OP_CUDA_KERNEL(
2121
ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, float>,
2222
ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, double>,
2323
ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int>,
24+
ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
2425
ops::LoadCombineOpKernel<paddle::platform::CUDADeviceContext, int64_t>);

paddle/fluid/operators/load_op.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,5 @@ REGISTER_OP_CUDA_KERNEL(
2020
load, ops::LoadOpKernel<paddle::platform::CUDADeviceContext, float>,
2121
ops::LoadOpKernel<paddle::platform::CUDADeviceContext, double>,
2222
ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int>,
23+
ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
2324
ops::LoadOpKernel<paddle::platform::CUDADeviceContext, int64_t>);

paddle/fluid/operators/save_combine_op.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,5 @@ REGISTER_OP_CUDA_KERNEL(
2121
ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, float>,
2222
ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, double>,
2323
ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int>,
24+
ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
2425
ops::SaveCombineOpKernel<paddle::platform::CUDADeviceContext, int64_t>);

paddle/fluid/operators/save_op.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ REGISTER_OP_CUDA_KERNEL(
2121
save, ops::SaveOpKernel<paddle::platform::CUDADeviceContext, float>,
2222
ops::SaveOpKernel<paddle::platform::CUDADeviceContext, double>,
2323
ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int>,
24+
ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int8_t>,
2425
ops::SaveOpKernel<paddle::platform::CUDADeviceContext, int64_t>,
2526
ops::SaveOpKernel<paddle::platform::CUDADeviceContext,
2627
paddle::platform::float16>);

0 commit comments

Comments
 (0)