More robust BroadcastStyle handling

[ToucheSir]

This makes use of the officially documented way to change the dimension of a BroadcastStyle. Ref. https://docs.julialang.org/en/v1/manual/interfaces/#writing-binary-broadcasting-rules:

Whenever you subtype AbstractArrayStyle, you also need to define rules for combining dimensionalities, by creating a constructor for your style that takes a Val(N) argument...

Which means we can rely on this being present for all GPU array types.

PR Checklist

• Tests are added
• Documentation, if applicable

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (06aba31) 96.40% compared to head (8a6bc82) 96.40%.

❗ Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

@@           Coverage Diff           @@
##             main      #43   +/-   ##
=======================================
  Coverage   96.40%   96.40%           
=======================================
  Files           4        4           
  Lines         139      139           
=======================================
  Hits          134      134           
  Misses          5        5           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[darsnack]

Failures appear unrelated