ptrendx commented on issue #19521:
URL:
https://github.com/apache/incubator-mxnet/issues/19521#issuecomment-728182152
Yes, currently the structure we have is
- `operator_name.cc` which contains operator definition (+ all the
infershape/type etc.) and `FCompute<cpu>`
- `operator_name.cu` which contains just `FCompute<gpu>`
We should change that to something like:
- `src/operator/operator_name.cc` which contains all the device independent
operator definition
- `src/operator_impl/cpu/operator_name.cc` which contains just
`FCompute<cpu>`
- `src/operator_impl/cuda/operator_name.cu` which contains just
`FCompute<gpu>`
This would make it possible to have a subgraph backend replace whatever they
need, as all the operator definitions would still exist. And I agree, together
with the external ops functionality we could make it so `libmxnet.so` contains
just the operator definitions, while separate `.so` would contains the actual
implementations for different platforms.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
