From 12b6158c5cbc10c56f935985e6f466c9867d9238 Mon Sep 17 00:00:00 2001 From: Min Xu <24926999+min-xu-ai@users.noreply.github.com> Date: Thu, 30 Sep 2021 17:33:08 -0700 Subject: [PATCH] [DOCS] Minor fix (#317) Co-authored-by: Min Xu --- python/src/functions.h | 4 ++-- python/triton/language/core.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/src/functions.h b/python/src/functions.h index 4d426890e..0f5a5c42f 100644 --- a/python/src/functions.h +++ b/python/src/functions.h @@ -314,7 +314,7 @@ std::string zeros_docstr = R"pbdoc( :param shape: Shape of the new array, e.g., (8, 16) or (8, ) :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 )pbdoc"; 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 slices, ir::builde } } return builder->create_reshape(self, shape); -} \ No newline at end of file +} diff --git a/python/triton/language/core.py b/python/triton/language/core.py index 6f192b2c6..29e867502 100644 --- a/python/triton/language/core.py +++ b/python/triton/language/core.py @@ -307,7 +307,7 @@ def zeros(shape, dtype, _builder=None): :param shape: Shape of the new array, e.g., (8, 16) or (8, ) :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 """ 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_i = off_i + (ij % size_g) new_j = (ij % size_gj) // size_g - return new_i, new_j \ No newline at end of file + return new_i, new_j