FluxML / FluxML/NNlib.jl

type-piracy of materialize

Open
#504 1 comment 0 reactions 0 assignees View on GitHub
CUDA good first issue
Dominant language
Julia
Stars
258
Forks
138
Avg merge
2d 1h
Merged PRs (30d)
3

Description

Looking at the SnoopCompile report, this is pirating the materialize function, causing latency problems after loading this package:
```
[1] materialize(bc::Base.Broadcast.Broadcasted{<:Any, <:Any, typeof(NNlib.relu), <:Tuple{CUDA.CuArray{<:Union{Float16, Float32, Float64}}}})
@ NNlibCUDA ~/.julia/packages/NNlibCUDA/C6t0p/src/cudnn/activations.jl:28
[2] materialize(bc::Base.Broadcast.Broadcasted{<:Any, <:Any, typeof(NNlib.elu), <:Tuple{CUDA.CuArray{<:Union{Float16, Float32, Float64}}}})
@ NNlibCUDA ~/.julia/packages/NNlibCUDA/C6t0p/src/cudnn/activations.jl:28
[3] materialize(bc::Base.Broadcast.Broadcasted{<:Any, <:Any, typeof(NNlib.σ), <:Tuple{CUDA.CuArray{<:Union{Float16, Float32, Float64}}}})
@ NNlibCUDA ~/.julia/packages/NNlibCUDA/C6t0p/src/cudnn/activations.jl:28
[4] materialize(bc::Base.Broadcast.Broadcasted{<:Any, <:Any, typeof(tanh), <:Tuple{CUDA.CuArray{<:Union{Float16, Float32, Float64}}}})
@ NNlibCUDA ~/.julia/packages/NNlibCUDA/C6t0p/src/cudnn/activations.jl:28
```
But if we look at the docs, the function for overloading is `broadcasted` instead:
https://docs.julialang.org/en/v1/manual/interfaces/#man-interfaces-broadcasting

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.