FluxML / FluxML/NNlib.jl

Add warnings/errors on CPU kernels when called with GPU arrays

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

Description

https://github.com/FluxML/model-zoo/issues/400 is but the most recent example of where this would help. Now that we have package extensions, this should be pretty trivial. In total there are probably less than a dozen conv + pooling functions to cover, I think.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reading the linked FluxML/model-zoo issue and locating the CPU convolution and pooling entry points in NNlib.jl. Check how package extensions distinguish CPU and GPU arrays, then cover the mentioned functions with warnings or errors for GPU-array calls; done means the behavior is applied consistently across the listed kernels and verified by tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
julia
Domain
machine-learning
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
42/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.