Skip to content

Conversation

@LittleHeroZZZX
Copy link
Contributor

@LittleHeroZZZX LittleHeroZZZX commented May 22, 2025

与主仓 PaddlePaddle/Paddle#72639 同步,修改 pool2d 的参数为std::vector<int64_t>

@paddle-bot
Copy link

paddle-bot bot commented May 22, 2025

Thanks for your contribution!

@LittleHeroZZZX LittleHeroZZZX changed the title [Custom Device] modefy pool2d signature to support int64 [Custom Device] modefy pool2d signature from int32 to int64 May 23, 2025
std::vector<int> strides_t =
std::vector<int>(strides_t_64.begin(), strides_t_64.end());
std::vector<int> paddings_t =
std::vector<int>(paddings_t_64.begin(), paddings_t_64.end());
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里如果不转,直接传给gcu 会有问题吗

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

可以不转

@xiaoguoguo626807 xiaoguoguo626807 merged commit 833ebc6 into PaddlePaddle:develop May 27, 2025
10 of 11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants