Commit d3c8256b by ziheng Committed by Tianqi Chen

Fix comment (#115)

parent e988e435
...@@ -284,16 +284,16 @@ constexpr const char* tvm_global_barrier_kinit = "tvm_global_barrier_kinit"; ...@@ -284,16 +284,16 @@ constexpr const char* tvm_global_barrier_kinit = "tvm_global_barrier_kinit";
/*! /*!
* \brief See pesudo code * \brief See pesudo code
* *
* Expr tvm_thread_allreduce(std::string op, Expr value, Expr cond, * Expr tvm_thread_allreduce(CommReducer combiner, Expr value, Expr cond,
* Var thread_idx1, thread_idx2...) { * Var thread_idx1, thread_idx2...) {
* // constraint by the other thread_idx remain the same. * // constraint by the other thread_idx remain the same.
* return reduce(op, value, cond, * return reduce(combiner, value, cond,
* over [thread_idx1, thread_idx2] passed by any caller) * over [thread_idx1, thread_idx2] passed by any caller)
* } * }
*/ */
constexpr const char* tvm_thread_allreduce = "tvm_thread_allreduce"; constexpr const char* tvm_thread_allreduce = "tvm_thread_allreduce";
/*! \brief The kind of structre field info */ /*! \brief The kind of structure field info */
enum TVMStructFieldKind : int { enum TVMStructFieldKind : int {
// array head address // array head address
kArrAddr, kArrAddr,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment