Skip to content
Unverified Commit f4b73fc7 authored by Kaushik Kulkarni's avatar Kaushik Kulkarni Committed by GitHub
Browse files

Broadcast binary ops with device scalars (#502)



* implement broadcasting array binary ops with device scalars

* test broadcasting array binary ops with device scalars

* re-add some asserts (better to be safe)

* be explicit in error msg

op -> operator

Co-authored-by: default avatarAndreas Klöckner <inform@tiker.net>

* array binops: include asserts on out.shape as well

* set default args more elegantly

* Array shape checks: save shapes in temporaries

Co-authored-by: default avatarAndreas Klöckner <inform@tiker.net>
parent a56281af
Pipeline #183419 passed with warnings with stage
in 17 minutes and 29 seconds
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment