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

2
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) {

2
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]