in reply to Re^2: Dataflow programming on CPU and GPU using AI::MXNet
in thread Dataflow programming on CPU and GPU using AI::MXNet
Lazy evaluation currently only happens with slice-like operations, but there are plans to switch to lazy evaluation in order to have not only loop-fusion by creating custom operations that would e.g. do the a*b + c*d with only one round-trip from RAM through the CPU to RAM, but also GPU processing. See https://github.com/PDLPorters/pdl/issues/349 for discussion and/or to participate!
|
---|
Replies are listed 'Best First'. | |
---|---|
Re^4: Dataflow programming on CPU and GPU using AI::MXNet
by etj (Priest) on Aug 04, 2024 at 23:58 UTC |