wkcn commented on a change in pull request #17270: Dynamic custom operator GPU
support
URL: https://github.com/apache/incubator-mxnet/pull/17270#discussion_r372699518
##########
File path: include/mxnet/lib_api.h
##########
@@ -203,6 +214,16 @@ enum MXDType {
kUNSET = 100,
};
+/*!
+ * \brief Context info passing from MXNet OpContext
+ * dev_type is string repr of supported context, currently only "cpu" and "gpu"
+ * dev_id is the device index where the tensor locates
+ */
+typedef struct {
+ std::string dev_type;
Review comment:
I see. Thank you!
----------------------------------------------------------------
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]
With regards,
Apache Git Services
