提交 7ca48f9c 编写于 作者: P phlrain

fix paddle throw type

上级 867fc053
...@@ -115,7 +115,8 @@ void EmbeddingGradKernel(const Context& ctx, ...@@ -115,7 +115,8 @@ void EmbeddingGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
...@@ -195,7 +196,8 @@ void EmbeddingSparseGradKernel(const Context& ctx, ...@@ -195,7 +196,8 @@ void EmbeddingSparseGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
...@@ -98,7 +98,8 @@ void EmbeddingKernel(const Context& ctx, ...@@ -98,7 +98,8 @@ void EmbeddingKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
...@@ -180,7 +180,8 @@ void SparseWeightEmbeddingGradKernel(const Context& ctx, ...@@ -180,7 +180,8 @@ void SparseWeightEmbeddingGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
...@@ -199,7 +200,8 @@ void SparseWeightEmbeddingSparseGradKernel(const Context& ctx, ...@@ -199,7 +200,8 @@ void SparseWeightEmbeddingSparseGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
...@@ -102,7 +102,8 @@ void SparseWeightEmbeddingKernel(const Context& ctx, ...@@ -102,7 +102,8 @@ void SparseWeightEmbeddingKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
...@@ -131,7 +131,8 @@ void EmbeddingGradKernel(const Context& ctx, ...@@ -131,7 +131,8 @@ void EmbeddingGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
...@@ -233,7 +234,8 @@ void EmbeddingSparseGradKernel(const Context& ctx, ...@@ -233,7 +234,8 @@ void EmbeddingSparseGradKernel(const Context& ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
...@@ -110,7 +110,8 @@ void EmbeddingKernel(const Context &ctx, ...@@ -110,7 +110,8 @@ void EmbeddingKernel(const Context &ctx,
} else if (input.dtype() == phi::DataType::INT64) { } else if (input.dtype() == phi::DataType::INT64) {
functor.template apply<int64_t>(); functor.template apply<int64_t>();
} else { } else {
PADDLE_THROW("emebdding input only support int32 and int64"); PADDLE_THROW(phi::errors::Unimplemented(
"emebdding input only support int32 and int64"));
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册