You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Calculates reduction step candidates for each reduction axis in a PrimFuncNode. General idea : use factor first, since it does not require extra boundary check. for large prime number, which is rare case, use power of 2.
83
-
84
-
:param node: The node for which to calculate reduction step candidates. It contains reduction axes (raxis)
85
-
with their domains (dom.extent).
86
-
:type node: PrimFuncNode
87
-
88
-
:returns: A dictionary mapping axis variable names to lists of step candidates. For each axis in the node,
89
-
this function calculates possible step sizes. For axes with a large prime domain, it uses powers of 2
90
-
as step candidates; for others, it uses all factors of the domain.
91
-
:rtype: Dict[str, List[int]]
92
-
93
-
94
67
95
68
.. py:method:: check_tile_shape_isvalid(td)
96
69
97
-
Checks if the tile shapes in the TileDict are valid for the nodes in this context.
98
-
99
-
Parameters:
100
-
- td (TileDict): The TileDict object containing tile shapes and other configurations.
101
-
102
-
Returns:
103
-
- bool: True if all tile shapes are valid, False otherwise.
104
-
105
-
106
70
107
71
.. py:method:: compute_node_stride_map(node, td)
108
72
109
-
Computes the stride map for a given node based on the TileDict configuration.
110
-
111
-
:param node: The node for which to compute the stride map.
112
-
:type node: PrimFuncNode
113
-
:param td: The TileDict object containing the tile configuration.
114
-
:type td: TileDict
115
-
116
-
:returns: A tuple of dictionaries containing the output strides and tensor strides.
117
-
:rtype: Tuple[Dict, Dict]
118
-
119
-
120
73
121
74
.. py:method:: plan_rasterization(td)
122
75
123
-
Plans the rasterization for the given TileDict. This function is not implemented yet.
124
-
125
-
:param td: The TileDict object to plan rasterization for.
126
-
:type td: TileDict
127
-
128
-
:raises RasterRationPlan: This function is not implemented yet.
<emclass="property"><spanclass="pre">class</span><spanclass="w"></span></em><spanclass="sig-prename descclassname"><spanclass="pre">tilelang.carver.arch.cdna.</span></span><spanclass="sig-name descname"><spanclass="pre">CDNA</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">target</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.arch.cdna.CDNA" title="Link to this definition">¶</a></dt>
<emclass="property"><spanclass="pre">class</span><spanclass="w"></span></em><spanclass="sig-prename descclassname"><spanclass="pre">tilelang.carver.arch.cpu.</span></span><spanclass="sig-name descname"><spanclass="pre">CPU</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">target</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.arch.cpu.CPU" title="Link to this definition">¶</a></dt>
<emclass="property"><spanclass="pre">class</span><spanclass="w"></span></em><spanclass="sig-prename descclassname"><spanclass="pre">tilelang.carver.arch.cuda.</span></span><spanclass="sig-name descname"><spanclass="pre">CUDA</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">target</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.arch.cuda.CUDA" title="Link to this definition">¶</a></dt>
Copy file name to clipboardExpand all lines: autoapi/tilelang/carver/roller/hint/index.html
+8-8Lines changed: 8 additions & 8 deletions
Original file line number
Diff line number
Diff line change
@@ -565,10 +565,10 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
565
565
<spanclass="sig-name descname"><spanclass="pre">compute_strides_from_shape</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">shape</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.roller.hint.Stride.compute_strides_from_shape" title="Link to this definition">¶</a></dt>
@@ -578,7 +578,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
578
578
<spanclass="sig-name descname"><spanclass="pre">compute_elements_from_shape</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">shape</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.roller.hint.Stride.compute_elements_from_shape" title="Link to this definition">¶</a></dt>
@@ -677,7 +677,7 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
677
677
<spanclass="sig-name descname"><spanclass="pre">get_tile</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">func</span></span></em><spanclass="sig-paren">)</span><aclass="headerlink" href="#tilelang.carver.roller.hint.TileDict.get_tile" title="Link to this definition">¶</a></dt>
<emclass="property"><spanclass="pre">property</span><spanclass="w"></span></em><spanclass="sig-name descname"><spanclass="pre">raxis_order</span></span><emclass="property"><spanclass="p"><spanclass="pre">:</span></span><spanclass="w"></span><spanclass="pre">tilelang.carver.roller.rasterization.List</span><spanclass="p"><spanclass="pre">[</span></span><spanclass="pre">int</span><spanclass="p"><spanclass="pre">]</span></span></em><aclass="headerlink" href="#tilelang.carver.roller.hint.Hint.raxis_order" title="Link to this definition">¶</a></dt>
949
+
<emclass="property"><spanclass="pre">property</span><spanclass="w"></span></em><spanclass="sig-name descname"><spanclass="pre">raxis_order</span></span><emclass="property"><spanclass="p"><spanclass="pre">:</span></span><spanclass="w"></span><spanclass="pre">List</span><spanclass="p"><spanclass="pre">[</span></span><spanclass="pre">int</span><spanclass="p"><spanclass="pre">]</span></span></em><aclass="headerlink" href="#tilelang.carver.roller.hint.Hint.raxis_order" title="Link to this definition">¶</a></dt>
<emclass="property"><spanclass="pre">property</span><spanclass="w"></span></em><spanclass="sig-name descname"><spanclass="pre">step</span></span><emclass="property"><spanclass="p"><spanclass="pre">:</span></span><spanclass="w"></span><spanclass="pre">tilelang.carver.roller.rasterization.List</span><spanclass="p"><spanclass="pre">[</span></span><spanclass="pre">int</span><spanclass="p"><spanclass="pre">]</span></span></em><aclass="headerlink" href="#tilelang.carver.roller.hint.Hint.step" title="Link to this definition">¶</a></dt>
959
+
<emclass="property"><spanclass="pre">property</span><spanclass="w"></span></em><spanclass="sig-name descname"><spanclass="pre">step</span></span><emclass="property"><spanclass="p"><spanclass="pre">:</span></span><spanclass="w"></span><spanclass="pre">List</span><spanclass="p"><spanclass="pre">[</span></span><spanclass="pre">int</span><spanclass="p"><spanclass="pre">]</span></span></em><aclass="headerlink" href="#tilelang.carver.roller.hint.Hint.step" title="Link to this definition">¶</a></dt>
0 commit comments