Skip to content

Commit 1186b2e

Browse files
Update docs
1 parent 3c483c7 commit 1186b2e

File tree

28 files changed

+173
-190
lines changed

28 files changed

+173
-190
lines changed

_sources/autoapi/tilelang/carver/arch/cdna/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,6 @@ Module Contents
3030
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
3131

3232

33-
Represents the architecture of a computing device, capturing various hardware specifications.
34-
35-
3633
.. py:attribute:: target
3734
3835

_sources/autoapi/tilelang/carver/arch/cpu/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,6 @@ Module Contents
3030
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
3131

3232

33-
Represents the architecture of a computing device, capturing various hardware specifications.
34-
35-
3633
.. py:attribute:: target
3734
3835

_sources/autoapi/tilelang/carver/arch/cuda/index.rst.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,9 +92,6 @@ Module Contents
9292
Bases: :py:obj:`tilelang.carver.arch.arch_base.TileDevice`
9393

9494

95-
Represents the architecture of a computing device, capturing various hardware specifications.
96-
97-
9895
.. py:attribute:: target
9996
10097

_sources/autoapi/tilelang/carver/roller/hint/index.rst.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -307,12 +307,12 @@ Module Contents
307307
308308
309309
.. py:property:: raxis_order
310-
:type: tilelang.carver.roller.rasterization.List[int]
310+
:type: List[int]
311311

312312

313313

314314
.. py:property:: step
315-
:type: tilelang.carver.roller.rasterization.List[int]
315+
:type: List[int]
316316

317317

318318

_sources/autoapi/tilelang/carver/roller/policy/tensorcore/index.rst.txt

Lines changed: 1 addition & 56 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,11 @@ Module Contents
3030

3131
.. py:data:: logger
3232
33-
.. py:class:: TensorCorePolicy(arch, tags = None)
33+
.. py:class:: TensorCorePolicy
3434
3535
Bases: :py:obj:`tilelang.carver.roller.policy.default.DefaultPolicy`
3636

3737

38-
Default Policy for fastdlight, a heuristic plan that tries to
39-
minimize memory traffic and maximize parallelism.for BitBLAS Schedule.
40-
41-
4238
.. py:attribute:: wmma_k
4339
:type: int
4440
:value: 16
@@ -65,67 +61,16 @@ Module Contents
6561

6662
.. py:method:: infer_node_smem_usage(td, node)
6763
68-
Infers the shared memory usage of a node given a TileDict configuration.
69-
70-
:param td: The TileDict object containing the tile configuration.
71-
:type td: TileDict
72-
:param node: The node for which to infer the shared memory usage.
73-
:type node: PrimFuncNode
74-
75-
:returns: The estimated amount of shared memory used by the node.
76-
:rtype: int
77-
78-
7964
8065
.. py:method:: get_node_reduce_step_candidates(node)
8166
82-
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-
9467
9568
.. py:method:: check_tile_shape_isvalid(td)
9669
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-
10670
10771
.. py:method:: compute_node_stride_map(node, td)
10872
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-
12073
12174
.. py:method:: plan_rasterization(td)
12275
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.
129-
130-
13176

_sources/autoapi/tilelang/jit/adapter/cython/adapter/index.rst.txt

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,18 @@ Module Contents
134134

135135

136136

137+
.. py:attribute:: static_strides_map
138+
:type: Optional[Dict[tvm.tir.Var, Tuple[int, List[Tuple[int, int]]]]]
139+
:value: None
140+
141+
142+
143+
.. py:attribute:: static_contiguous_list
144+
:type: Optional[List[tvm.tir.Var]]
145+
:value: None
146+
147+
148+
137149
.. py:attribute:: buffer_device_map
138150
:type: Optional[Dict[tvm.tir.Var, Tuple[int, torch.device]]]
139151
:value: None

_sources/autoapi/tilelang/jit/adapter/dlpack/index.rst.txt

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,12 +20,8 @@ Classes
2020
Module Contents
2121
---------------
2222

23-
.. py:class:: TorchDLPackKernelAdapter(mod, params, result_idx)
23+
.. py:class:: TorchDLPackKernelAdapter
2424
2525
Bases: :py:obj:`tilelang.jit.adapter.base.BaseKernelAdapter`
2626

2727

28-
Helper class that provides a standard way to create an ABC using
29-
inheritance.
30-
31-

_sources/autoapi/tilelang/jit/adapter/nvrtc/adapter/index.rst.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,6 @@ Module Contents
8080
8181
8282
.. py:attribute:: target
83-
:value: None
84-
8583
8684
8785
.. py:attribute:: verbose

_sources/autoapi/tilelang/jit/adapter/wrapper/index.rst.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -493,6 +493,9 @@ Module Contents
493493
.. py:property:: prim_func
494494
495495
496+
.. py:property:: device_func
497+
498+
496499
.. py:class:: TLNVRTCSourceWrapper(scheduled_ir_module, source, target, device_mod = None, host_mod = None, pass_configs = None)
497500
498501
Bases: :py:obj:`TLCUDASourceWrapper`

_sources/autoapi/tilelang/language/proxy/index.rst.txt

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ Classes
2525
tilelang.language.proxy.BufferProxy
2626
tilelang.language.proxy.BaseTensorProxy
2727
tilelang.language.proxy.TensorProxy
28+
tilelang.language.proxy.StridedTensorProxy
2829
tilelang.language.proxy.FragmentBufferProxy
2930
tilelang.language.proxy.SharedBufferProxy
3031
tilelang.language.proxy.LocalBufferProxy
@@ -54,7 +55,7 @@ Module Contents
5455
.. py:method:: __getitem__(keys)
5556
5657
57-
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32')
58+
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32', strides = None)
5859
5960
Create a buffer from a pointer, shape, and data type.
6061

@@ -96,7 +97,7 @@ Module Contents
9697
.. py:method:: __getitem__(keys)
9798
9899
99-
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32')
100+
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32', strides = None)
100101
101102
Create a buffer from a pointer, shape, and data type.
102103

@@ -116,7 +117,23 @@ Module Contents
116117
Main tensor proxy class for global scope buffers.
117118

118119
This class implements the default tensor proxy with global memory scope,
119-
inheriting all functionality from BaseTensorProxy without modifications.
120+
the tensor should be by default contiguous.
121+
122+
123+
.. py:method:: __call__(shape, dtype = 'float32', data=None)
124+
125+
126+
.. py:class:: StridedTensorProxy
127+
128+
Bases: :py:obj:`BaseTensorProxy`
129+
130+
131+
Main tensor proxy class for global scope buffers, with strides supported.
132+
133+
This class implements the default tensor proxy with global memory scope, with the stride information required.
134+
135+
136+
.. py:method:: __call__(shape, strides, dtype = 'float32')
120137
121138
122139
.. py:class:: FragmentBufferProxy
@@ -182,7 +199,7 @@ Module Contents
182199
.. py:method:: __setitem__(key, value)
183200
184201
185-
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32')
202+
.. py:method:: from_ptr(pointer_var, shape, dtype = 'float32', strides = None)
186203
:classmethod:
187204

188205

@@ -202,5 +219,5 @@ Module Contents
202219
:rtype: PrimExpr
203220

204221

205-
.. py:function:: make_tensor(ptr, shape, dtype = 'float32')
222+
.. py:function:: make_tensor(ptr, shape, dtype = 'float32', strides = None)
206223

0 commit comments

Comments
 (0)