Fix Linear layer bias gradient computation; add size checks to CUDA functions#170

Merged

drahnr merged 4 commits intofff-rs:masterfrom hweom:copy-fix

Aug 15, 2022

+61-18

Commits on Jul 28, 2022

Assert the correct tensor sizes in copy() and gemm(); fix related Linear logic

Mikhail Balakhno
authored and
Mikhail Balakhno
committed
Check output matrix dims in GEMM; fix corresponding Linear layer logic

Mikhail Balakhno
authored and
Mikhail Balakhno
committed

Commits on Aug 11, 2022

Update coaster-blas/src/frameworks/cuda/helper.rs

hweom
and
drahnr
authored

Commits on Aug 14, 2022

Merge branch 'master' into copy-fix

Mikhail Balakhno
authored and
Mikhail Balakhno
committed