Skip to content

Commit

Permalink
try to fix compile on windows
Browse files Browse the repository at this point in the history
  • Loading branch information
SylarTiaNII committed Jan 16, 2024
1 parent 49a42be commit ce68851
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions paddle/fluid/operators/custom_device_common_op_registry.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1383,24 +1383,24 @@ void RegisterCustomDeviceCommonKernel(const std::string& dev_type) {
paddle::operators::CSplitOpCustomDeviceKernel<
paddle::platform::CustomDeviceContext,
paddle::platform::float16>);
REGISTER_OP_CUSTOM_DEVICE_KERNEL(
c_embedding,
device_type,
paddle::operators::CEmbeddingOpCustomDeviceKernel<
paddle::platform::CustomDeviceContext,
float>,
paddle::operators::CEmbeddingOpCustomDeviceKernel<
paddle::platform::CustomDeviceContext,
paddle::platform::float16>);
REGISTER_OP_CUSTOM_DEVICE_KERNEL(
c_embedding_grad,
device_type,
paddle::operators::CEmbeddingGradOpCustomDeviceKernel<
paddle::platform::CustomDeviceContext,
float>,
paddle::operators::CEmbeddingGradOpCustomDeviceKernel<
paddle::platform::CustomDeviceContext,
paddle::platform::float16>);
// REGISTER_OP_CUSTOM_DEVICE_KERNEL(
// c_embedding,
// device_type,
// paddle::operators::CEmbeddingOpCustomDeviceKernel<
// paddle::platform::CustomDeviceContext,
// float>,
// paddle::operators::CEmbeddingOpCustomDeviceKernel<
// paddle::platform::CustomDeviceContext,
// paddle::platform::float16>);
// REGISTER_OP_CUSTOM_DEVICE_KERNEL(
// c_embedding_grad,
// device_type,
// paddle::operators::CEmbeddingGradOpCustomDeviceKernel<
// paddle::platform::CustomDeviceContext,
// float>,
// paddle::operators::CEmbeddingGradOpCustomDeviceKernel<
// paddle::platform::CustomDeviceContext,
// paddle::platform::float16>);

REGISTER_OP_CUSTOM_DEVICE_KERNEL(
c_softmax_with_cross_entropy,
Expand Down

0 comments on commit ce68851

Please sign in to comment.