20 #ifndef MXNET_IMPERATIVE_H_ 21 #define MXNET_IMPERATIVE_H_ 33 #include <unordered_map> 63 grad_req(
kNullOp), fresh_out_grad(false) {}
66 if (node ==
nullptr || node->info.empty())
return;
73 return dmlc::get<AGInfo>(node->info);
77 node->info.construct<
AGInfo>();
82 return arr.entry_.
node ==
nullptr || arr.entry_.
node->info.empty();
103 return is_recording_;
107 bool old = is_recording_;
115 if (is_np_shape_global_) {
118 return is_np_shape_thread_local_ ? 1 : 0;
126 is_np_shape_global_ =
true;
127 is_np_shape_thread_local_ =
true;
130 is_np_shape_thread_local_ =
true;
133 is_np_shape_global_ =
false;
134 is_np_shape_thread_local_ =
false;
141 const std::vector<NDArray*>& inputs,
142 const std::vector<NDArray*>&
outputs,
144 std::vector<bool>* p_save_inputs =
nullptr,
145 std::vector<bool>* p_save_outputs =
nullptr);
149 const std::vector<NDArray*>& inputs,
150 const std::vector<NDArray*>& outputs);
154 const std::vector<NDArray*>& inputs,
155 const std::vector<NDArray*>& outputs,
156 const std::vector<OpReqType>& req,
161 const std::vector<uint32_t>& grad_reqs,
162 const std::vector<NDArray*>& gradients);
164 std::vector<NDArray*>
Backward(
const std::vector<NDArray*>& outputs,
165 const std::vector<NDArray*>& ograds,
166 const std::vector<NDArray*>& variables,
167 bool is_train,
bool retain_graph,
173 return dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_INFERENCE",
true);
177 return dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_TRAIN",
true);
181 return dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_MAX_NODE_TRAIN_FWD",
182 dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_MAX_NODE_TRAIN", 15));
186 return dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_MAX_NODE_TRAIN_BWD",
187 dmlc::GetEnv(
"MXNET_EXEC_BULK_EXEC_MAX_NODE_TRAIN", 15));
198 void GetBackwardDependency(
200 uint32_t num_inputs, uint32_t num_outputs,
201 std::vector<bool> *p_save_inputs,
202 std::vector<bool> *p_save_outputs);
204 #if DMLC_CXX11_THREAD_LOCAL 205 static thread_local
bool is_train_;
206 static thread_local
bool is_recording_;
209 static thread_local
bool is_np_shape_thread_local_;
211 static MX_THREAD_LOCAL
bool is_train_;
212 static MX_THREAD_LOCAL
bool is_recording_;
215 static MX_THREAD_LOCAL
bool is_np_shape_thread_local_;
217 bool is_np_shape_global_{
false};
219 std::atomic<uint64_t> node_count_{0};
221 std::atomic<uint64_t> variable_count_{0};
223 int backward_bulk_size_{0};
227 #endif // MXNET_IMPERATIVE_H_ Definition: imperative.h:48
bool is_recording() const
whether operator recording is on.
Definition: imperative.h:102
static bool IsNone(const NDArray &arr)
Definition: imperative.h:81
static int BulkExecMaxNodeTrainFwd()
The max number of op nodes in a bulk during forward pass of training.
Definition: imperative.h:180
static AGInfo & Create(const nnvm::NodePtr &node)
Definition: imperative.h:76
Definition: imperative.h:48
bool is_training() const
whether operator recording is on.
Definition: imperative.h:92
no operation, do not write anything
Definition: op_attr_types.h:47
bool set_is_training(bool is_train)
turn on or turn off operator recording for autograd.
Definition: imperative.h:96
The attributes of the current operation node. Usually are additional parameters like axis...
Definition: node.h:120
namespace of mxnet
Definition: base.h:89
std::vector< NDArray * > Backward(const std::vector< NDArray * > &outputs, const std::vector< NDArray * > &ograds, const std::vector< NDArray * > &variables, bool is_train, bool retain_graph, bool create_graph)
compute the gradient of outputs w.r.t variables.
static void Clear(const nnvm::NodePtr &node)
Definition: imperative.h:65
OpStatePtr Invoke(const Context &default_ctx, const nnvm::NodeAttrs &attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs)
std::shared_ptr< Node > NodePtr
we always used NodePtr for a reference pointer to the node, so this alias can be changed in case...
Definition: node.h:49
AGInfo()
Definition: imperative.h:62
DispatchMode
the dispatch mode of the operator
Definition: op_attr_types.h:122
NodePtr node
the source node of this data
Definition: node.h:76
std::vector< NDArray > outputs
Definition: imperative.h:58
int is_np_shape() const
return current numpy compatibility status, GlobalOn(2), ThreadLocalOn(1), Off(0). ...
Definition: imperative.h:114
Definition: imperative.h:53
bool set_is_recording(bool is_recording)
turn on or turn off operator recording for autograd.
Definition: imperative.h:106
bool fresh_out_grad
Definition: imperative.h:60
OpStatePtr state
Definition: imperative.h:57
std::vector< NDArray > out_grads
Definition: imperative.h:59
void RecordOp(nnvm::NodeAttrs &&attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs, const OpStatePtr &state=OpStatePtr(), std::vector< bool > *p_save_inputs=nullptr, std::vector< bool > *p_save_outputs=nullptr)
to record operator, return corresponding node.
OpReqType grad_req
Definition: imperative.h:56
Context ctx
Definition: imperative.h:55
Configuation of nnvm as well as basic data structure.
OpReqType
operation request type to Forward and Backward
Definition: op_attr_types.h:45
OpStatePtr InvokeOp(const Context &ctx, const nnvm::NodeAttrs &attrs, const std::vector< NDArray * > &inputs, const std::vector< NDArray * > &outputs, const std::vector< OpReqType > &req, const DispatchMode dispatch_mode, OpStatePtr state=OpStatePtr())
static bool IsVariable(const nnvm::NodePtr &node)
Definition: imperative.h:85
runtime functions for NDArray
Definition: imperative.h:50
Definition: imperative.h:48
static AGInfo & Get(const nnvm::NodePtr &node)
Definition: imperative.h:72
Operator information structor.
static bool PreferBulkExecTrain()
Should op execution bulking be employed during training.
Definition: imperative.h:176
Symbolic graph construction API.
static int BulkExecMaxNodeTrainBwd()
The max number of op nodes in a bulk during backward pass of training.
Definition: imperative.h:185
bool set_is_np_shape(int is_np_shape)
specify numpy compatibility off, thread local on or global on.
Definition: imperative.h:121
static bool PreferBulkExecInference()
Should op execution bulking be employed during inference.
Definition: imperative.h:172
void MarkVariables(const std::vector< NDArray * > &variables, const std::vector< uint32_t > &grad_reqs, const std::vector< NDArray * > &gradients)
mark variables for computing gradients.
Context information about the execution environment.
Definition: base.h:102
ndarray interface
Definition: ndarray.h:82
NumpyShape
there are three numpy shape flags based on priority. GlobalOn turn on numpy shape flag globally...
Definition: imperative.h:48
Operator state. This is a pointer type, its content is mutable even if OpStatePtr is const...
Definition: op_attr_types.h:148