name rule for debugging purposes

patrick-kidger / quax

Multiple dispatch over abstract array types in JAX.

Apache License 2.0

100 stars 2 forks source link

name rule for debugging purposes #1

Closed nstarman closed 9 months ago

nstarman commented 9 months ago

Very cool package! When I was debugging a trace I found it hard to identify the problematic primitive when the rule was just called existing_rule. This PR just renames the rule to identify the relevant primitive.

patrick-kidger commented 9 months ago

LGTM! The test failure looks like an unrelated flake, which I've just fixed separately.