Skip to content

Conversation

@donjuanplatinum
Copy link
Contributor

No description provided.

@ivarflakstad
Copy link
Member

ivarflakstad commented Oct 22, 2025

Nice!
If you want this to work on cuda / metal you'll have to add padding functionality to the actual kernels as well.
You can see the cuda kernel here and the metal kernel here.
And update the code where the kernels are called of course, like here for metal.

@donjuanplatinum
Copy link
Contributor Author

Nice! If you want this to work on cuda / metal you'll have to add padding functionality to the actual kernels as well. You can see the cuda kernel here and the metal kernel here. And update the code where the kernels are called of course, like here for metal.

I have modified the code :)

…e_padding \n2. combine the new call_pool2d_with_padding into original max_pool2d function \n3. change the metal_backend
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants