mxnet-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From Tianqi Chen <>
Subject Re: [apache/incubator-mxnet] [RFC][mxnet 2.0][item 10.1] MXNet Imperative Op Invocation Overhead (#17097)
Date Sun, 22 Dec 2019 05:32:42 GMT
The following fast-path can be addressed in the TVM FFI:
- ```tuple``, ```list``` via translation in python/cython side (see benchmark above)
- ```str``` is already fast (seem benchmark above)
- ```Context``` can be quite fast if the object is a tvm object, around same mangitude as
passing NDArray
- ```np.dtype``` can get around by by str conversion, or introduce a type structure, TVM FFI
support DLDataType natively
- None: natively supported by FFI

The following items needs to be discussed
- py_slice, Ellipsis Can be supported by adding pyobject support, however that introduces
dispatching into the FFI layer(making the function not accessible to other language frontends).
Would be interesting to discuss alternatives

You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
  • Unnamed multipart/alternative (inline, 7-Bit, 0 bytes)
View raw message