tilelang.language.tir.ir module#

tilelang.language.tir.ir.grid(*extents: PrimExpr) → ForFrame#

The grid For statement.

Parameters:

extents (PrimExpr) – The extents of the iteration.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame

tilelang.language.tir.ir.parallel(start: PrimExpr, stop: Optional[PrimExpr] = None, *, annotations: Optional[Dict[str, Any]] = None) → ForFrame#

The parallel For statement.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • annotations (Dict[str, Any]) – The optional annotations of the For statement.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame

tilelang.language.tir.ir.serial(start: PrimExpr, stop: Optional[PrimExpr] = None, *, annotations: Optional[Dict[str, Any]] = None) → ForFrame#

The serial For statement.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • annotations (Dict[str, Any]) – The optional annotations of the For statement.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame

tilelang.language.tir.ir.thread_binding(start: PrimExpr, stop: Optional[PrimExpr] = None, thread: Optional[str] = None, *, annotations: Optional[Dict[str, Any]] = None) → ForFrame#

The thread-binding For statement.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • thread (str) – The thread for loop variable to bind.

  • annotations (Dict[str, Any]) – The optional annotations of the For statement.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame

tilelang.language.tir.ir.unroll(start: PrimExpr, stop: Optional[PrimExpr] = None, *, annotations: Optional[Dict[str, Any]] = None) → ForFrame#

The unrolled For statement.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • annotations (Dict[str, Any]) – The optional annotations of the For statement.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame

tilelang.language.tir.ir.vectorized(start: PrimExpr, stop: Optional[PrimExpr] = None, *, annotations: Optional[Dict[str, Any]] = None) → ForFrame#

The vectorized For statement.

Parameters:

  • start (PrimExpr) – The minimum value of iteration.

  • stop (PrimExpr) – The maximum value of iteration.

  • annotations (Dict[str, Any]) – The optional annotations of the For statement.

Returns:

res – The ForFrame.

Return type:

frame.ForFrame