[DOCS] Minor fix (#317)

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

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, )
: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<py::object> slices, ir::builde
}
}
return builder->create_reshape(self, shape);
}
}