From 8a100774e58c8e4bbab5ff389bab78bd91dbd4e3 Mon Sep 17 00:00:00 2001
From: lilong12 <lilong12@baidu.com>
Date: Tue, 10 May 2022 14:22:14 +0800
Subject: [PATCH] add int8 for cast (#42634)

---
 paddle/fluid/operators/cast_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/cast_op.cc b/paddle/fluid/operators/cast_op.cc
index bc6cf9d831f..76e0f23df21 100644
--- 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>>,
-- 
GitLab