add int8 for cast (#42634)

8a100774 · lilong12 · GitHub · 6ac08db5 · 8a100774
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/cast_op.cc paddle/fluid/operators/cast_op.cc +1 -1

未找到文件。
--- a/paddle/fluid/operators/cast_op.cc
+++ b/paddle/fluid/operators/cast_op.cc
@@ -156,7 +156,7 @@ REGISTER_OP_CPU_KERNEL(
    ops::CastOpKernel<CPU, double>, ops::CastOpKernel<CPU, int>,
    ops::CastOpKernel<CPU, int64_t>, ops::CastOpKernel<CPU, int>,
    ops::CastOpKernel<CPU, int16_t>, ops::CastOpKernel<CPU, bool>,
-    ops::CastOpKernel<CPU, uint8_t>,
+    ops::CastOpKernel<CPU, uint8_t>, ops::CastOpKernel<CPU, int8_t>,
    ops::CastOpKernel<CPU, paddle::platform::float16>,
    ops::CastOpKernel<CPU, paddle::platform::bfloat16>,
    ops::CastOpKernel<CPU, paddle::platform::complex<float>>,