Add an attribute to FunctionNode to indicate it supports arrays on different devices #4459
Labels
cat:feature
Implementation that introduces new interfaces.
prio:low
Low priority. Nice-to-have.
roadmap
Included in future roadmap.
This idea came from an offline discussion related to #454.
Currently only array types are checked in FunctionNode inputs, but not their devices.
If the arrays are on different GPU devices, it causes an error that comes from CuPy, not easy to investigate the fault. It's better for users if Chainer can catch such error earlier and present more easy-to-understand error message.
On the other hand, there may exist (user-defined) functions that support arrays on multiple GPU devices.
The proposed solution is:
The text was updated successfully, but these errors were encountered: