EnzymeAD / EnzymeAD/Enzyme

Allow non-vectorized function arguments in Batch

Open
#3,016 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
LLVM
Stars
1.7k
Forks
188
Avg merge
1d 22h
Merged PRs (30d)
26

Description

Do I understand correctly that`BatchOp` assumes all function arguments should be vectorized?

If so, can we allow arguments that aren't, a bit like the inactive args in grad (I think)? I mainly want it because I'm passing values to my function that were intended to be captured by the function, but since this isn't possible, I pass them as arguments. Not having to broadcast them first is convenient and intuitive. More importantly for most people, perhaps, is that it's more efficient, because we don't broadcast values prematurely. I may want to do long calculations on the values that aren't vectorized _before_ they interact with vectorized values, and doing those calculations vectorized is a waste.

Contributor guide

Open the contributing guide

Research direction

Start by reading the BatchOp behavior and comparing it with the inactive arguments in grad. Determine how non-vectorized arguments should be represented and propagated without premature broadcasting, then add coverage showing that long calculations on those arguments remain non-vectorized before interacting with vectorized values.

Written by the indexing model from the issue text.

Assessment

Domain
compilers
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.