github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[DOCS] Fix typos (#2097)

Browse Source
This commit is contained in:
danny.jang
2023-08-13 22:52:43 +09:00
committed by GitHub
parent 98372f46d3
commit cd114c0bd5
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/triton/language/random.py
View File

@@ -55,7 +55,7 @@ def randint(seed, offset, n_rounds: tl.constexpr = N_ROUNDS_DEFAULT):
using `randint4x` is likely to be faster than calling `randint` 4 times.
:param seed: The seed for generating random numbers.
:param offsets: The offsets to generate random numbers for.
:param offset: The offsets to generate random numbers for.
"""
ret, _, _, _ = randint4x(seed, offset, n_rounds)
return ret
@@ -120,7 +120,7 @@ def rand(seed, offset, n_rounds: tl.constexpr = N_ROUNDS_DEFAULT):
def rand4x(seed, offsets, n_rounds: tl.constexpr = N_ROUNDS_DEFAULT):
"""
Given a :code:`seed` scalar and an :code:`offsets` block,
returns a 4 blocks of random :code:`float32` in :math:`U(0, 1)`.
returns 4 blocks of random :code:`float32` in :math:`U(0, 1)`.
:param seed: The seed for generating random numbers.
:param offsets: The offsets to generate random numbers for.
@@ -167,7 +167,7 @@ def randn(seed, offset, n_rounds: tl.constexpr = N_ROUNDS_DEFAULT):
def randn4x(seed, offset, n_rounds: tl.constexpr = N_ROUNDS_DEFAULT):
"""
Given a :code:`seed` scalar and an :code:`offset` block,
returns a 4 blocks of random :code:`float32` in :math:`\\mathcal{N}(0, 1)`.
returns 4 blocks of random :code:`float32` in :math:`\\mathcal{N}(0, 1)`.
:param seed: The seed for generating random numbers.
:param offsets: The offsets to generate random numbers for.

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

@@ -192,7 +192,7 @@ def truediv(input: tl.tensor,
elif input_scalar_ty.is_int() and other_scalar_ty.is_int():
input = cast(input, tl.float32, builder)
other = cast(other, tl.float32, builder)
# float / float (cast to highest exponent type)
# float / float (cast to the highest exponent type)
elif input_scalar_ty.is_floating() and other_scalar_ty.is_floating():
if input_scalar_ty.fp_mantissa_width > other_scalar_ty.fp_mantissa_width:
other = cast(other, input_scalar_ty, builder)
@@ -1364,7 +1364,7 @@ def associative_scan(
def _check_dtype(dtypes: List[str]) -> T:
"""
We following libdevice's convention to check accepted data types for math functions.
We're following libdevice's convention to check accepted data types for math functions.
It is not a good practice to support all data types as accelerators/GPUs don't support
many float16 and bfloat16 math operations.
We should let the users know that they are using and invoke explicit cast to convert