Skip to content
This repository has been archived by the owner on Nov 4, 2024. It is now read-only.

handle bitstypes and wrapped arrays in isleaf #88

Merged
merged 7 commits into from
Oct 22, 2024

Conversation

CarloLucibello
Copy link
Contributor

@CarloLucibello CarloLucibello commented Oct 20, 2024

Porting over some desirable behavior from Flux.
Make MLDataDevices.isleaf(x::AbstractArray{T}) = true when T is a bitstype. Also, make sure wrapped arrays are handled correctly as discussed in FluxML/Flux.jl#2432 and FluxML/Flux.jl#2436

@CarloLucibello
Copy link
Contributor Author

ready to go

Project.toml Outdated Show resolved Hide resolved
@CarloLucibello
Copy link
Contributor Author

merge?

@avik-pal
Copy link
Member

avik-pal commented Oct 22, 2024 via email

@avik-pal avik-pal merged commit e9a2ed7 into LuxDL:main Oct 22, 2024
25 of 37 checks passed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants