gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[DOCS] Minor fix (#317)

Browse Source 
Co-authored-by: Min Xu <min.xu.public@gmail.com>
This commit is contained in:
Min Xu
2021-09-30 17:33:08 -07:00
committed by GitHub
parent b352b16567
commit 12b6158c5c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/functions.h
View File

@@ -314,7 +314,7 @@ std::string zeros_docstr = R"pbdoc(
:param shape: Shape of the new array, e.g., (8, 16) or (8, ) :param shape: Shape of the new array, e.g., (8, 16) or (8, )
:type shape: tuple of ints :type shape: tuple of ints
:param dtype: Data-type of the new array, e.g., triton.float16 :param dtype: Data-type of the new array, e.g., tl.float16
:type dtype: triton.ir.dtype :type dtype: triton.ir.dtype
)pbdoc"; )pbdoc";
ir::value *zeros(ir::type::block_shapes_t shape, type_code _dtype, ir::builder *builder) { ir::value *zeros(ir::type::block_shapes_t shape, type_code _dtype, ir::builder *builder) {
@@ -673,4 +673,4 @@ ir::value *subscript(ir::value *self, std::vector<py::object> slices, ir::builde
} }
} }
return builder->create_reshape(self, shape); return builder->create_reshape(self, shape);
} }

4
python/triton/language/core.py
View File

@@ -307,7 +307,7 @@ def zeros(shape, dtype, _builder=None):
:param shape: Shape of the new array, e.g., (8, 16) or (8, ) :param shape: Shape of the new array, e.g., (8, 16) or (8, )
:type shape: tuple of ints :type shape: tuple of ints
:param dtype: Data-type of the new array, e.g., :code:`triton.float16` :param dtype: Data-type of the new array, e.g., :code:`tl.float16`
:type dtype: DType :type dtype: DType
""" """
shape = [int(x.handle) if isinstance(x, block) else x for x in shape] shape = [int(x.handle) if isinstance(x, block) else x for x in shape]
@@ -730,4 +730,4 @@ def swizzle2d(i, j, size_i, size_j, size_g):
# new row and column indices # new row and column indices
new_i = off_i + (ij % size_g) new_i = off_i + (ij % size_g)
new_j = (ij % size_gj) // size_g new_j = (ij % size_gj) // size_g
return new_i, new_j return new_i, new_j