mxnet-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From JackieWu <notificati...@github.com>
Subject Re: [apache/incubator-mxnet] [RFC] Custom Operator Part 2 (#17006)
Date Sun, 08 Dec 2019 00:13:53 GMT
Hi @samskalicky , thank you for the contribution!
I have several suggestions.

- custom GPU operators
  1. Provide CUDA stream in `OpResource`.
  2. Share the same function on CPU and GPU.
      Users can discriminate the context by `MXTensor::dltensor::ctx`
- Call framework specific math helper
  It is important for a custom operator. Users may call gemm, even convolution op in custom
op.

Thanks.

-- 
You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
https://github.com/apache/incubator-mxnet/issues/17006#issuecomment-562898682
Mime
  • Unnamed multipart/alternative (inline, 7-Bit, 0 bytes)
View raw message