FluxML / FluxML/NNlib.jl

warn on `batched_mul_generic!`?

Open
#478 2 comments 1 reaction 0 assignees View on GitHub
performance
Dominant language
Julia
Stars
258
Forks
138
Avg merge
2d 1h
Merged PRs (30d)
3

Description

The `batched_mul_generic!` is really slow on both cpu and gpu, but somehow it isn't something easy to find. It might worth displaying a one time warning like the cuda scalar indexing warning.

Contributor guide

No contributing guide indexed for this repository

Research direction

No files or tests are named. Start by locating `batched_mul_generic!` and the existing CUDA scalar-indexing warning, then determine how warnings are emitted across CPU and GPU backends. Done means users receive a one-time warning that this generic batched multiplication path is slow.

Written by the indexing model from the issue text.

Assessment

Tech stack
julia
Domain
machine-learning, performance
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.