[DOCS] Minor fix (#317)
Co-authored-by: Min Xu <min.xu.public@gmail.com>
This commit is contained in:
@@ -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);
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user