You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
The reason will be displayed to describe this comment to others. Learn more.
Hi @CarloLucibello, thanks for starting the MetalExt for Flux! So far so good. I think a good next step can be to get the Conv() and ConvTranspose() layers working with Metal gpu. It seems there's an issue with scalar indexing while using im2col!() function.
24b1eb2There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@JuliaRegistrator register
24b1eb2There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Registration pull request created: JuliaRegistries/General/85780
After the above pull request is merged, it is recommended that a tag is created on this repository for the registered package version.
This will be done automatically if the Julia TagBot GitHub Action is installed, or can be done manually through the github interface, or via:
24b1eb2There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @CarloLucibello, thanks for starting the MetalExt for Flux! So far so good. I think a good next step can be to get the Conv() and ConvTranspose() layers working with Metal gpu. It seems there's an issue with scalar indexing while using im2col!() function.
24b1eb2There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for testing! Can you open separate issues for the different missing functionalities?
24b1eb2There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done it here #2278. I'll be creating more issues, if there's any, as I keep testing this. Thanks!