Skip to content

Commit a302e0f

Browse files
author
Siyuan Feng
authored
[TIR] Enhance Python Type Annotations for TIR stmt (#16076)
This PR enhances the Python annotations for the TIR stmt, adding class member variables annotations.
1 parent 03f7b3b commit a302e0f

5 files changed

Lines changed: 233 additions & 89 deletions

File tree

include/tvm/tir/stmt.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ class LetStmtNode : public StmtNode {
6868
public:
6969
/*! \brief The variable. */
7070
Var var;
71-
/*! \brief The value to be binded. */
71+
/*! \brief The value to be bound. */
7272
PrimExpr value;
7373
/*! \brief The body block. */
7474
Stmt body;
@@ -876,7 +876,7 @@ class SeqStmt : public Stmt {
876876
};
877877

878878
/*!
879-
* \brief IfThenElse statment.
879+
* \brief IfThenElse statement.
880880
*/
881881
class IfThenElseNode : public StmtNode {
882882
public:
@@ -951,7 +951,7 @@ enum class ForKind : int {
951951
};
952952

953953
/*!
954-
* \brief A for loop, with poissible type annotations.
954+
* \brief A for loop, with possible type annotations.
955955
*
956956
* \code
957957
*
@@ -1388,7 +1388,7 @@ namespace attr {
13881388
constexpr const char* thread_extent = "thread_extent";
13891389
/*! \brief Mark launching of a virtual thread. */
13901390
constexpr const char* virtual_thread = "virtual_thread";
1391-
/*! \brief Mark region is processed by a co-proccesor */
1391+
/*! \brief Mark region is processed by a co-processor */
13921392
constexpr const char* coproc_scope = "coproc_scope";
13931393
/*!
13941394
* \brief Mark region creates coprocessor micro ops,

0 commit comments

Comments
 (0)