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

[DOC] Fix a typo in where's description (#1286)

Browse Source 
Co-authored-by: Alexander Zinoviev <zinoviev@google.com>
This commit is contained in:
Alexander Zinoviev
2023-03-06 14:38:03 -08:00
committed by GitHub
parent ff94e34430
commit 5e92a66267
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/language/core.py
View File

@@ -1027,7 +1027,7 @@ def where(condition, x, y, _builder=None):
If you want to avoid unintended memory operations, use the :code:`mask` arguments in `triton.load` and `triton.store` instead.
The shape of :code:`x` and :code:`y` are both broadcast to the shape of :code:`condition`.
:code:`x` and :code:`y` must have the data type.
:code:`x` and :code:`y` must have the same data type.
:param condition: When True (nonzero), yield x, otherwise yield y.
:type condition: Block of triton.bool