Skip to content

Commit ef319c8

Browse files
authored
Fix formatting in docs for triton.language.dot (#5020)
1 parent 018c139 commit ef319c8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/triton/language/core.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1523,9 +1523,9 @@ def dot(input, other, acc=None, input_precision=None, allow_tf32=None, max_num_i
15231523
where the first dimension of each block represents the batch dimension.
15241524
15251525
:param input: The first tensor to be multiplied.
1526-
:type input: 2D or 3D tensor of scalar-type in {:code:`int8`, :code: `float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
1526+
:type input: 2D or 3D tensor of scalar-type in {:code:`int8`, :code:`float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
15271527
:param other: The second tensor to be multiplied.
1528-
:type other: 2D or 3D tensor of scalar-type in {:code:`int8`, :code: `float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
1528+
:type other: 2D or 3D tensor of scalar-type in {:code:`int8`, :code:`float8_e5m2`, :code:`float16`, :code:`bfloat16`, :code:`float32`}
15291529
:param acc: The accumulator tensor. If not None, the result is added to this tensor.
15301530
:type acc: 2D or 3D tensor of scalar-type in {:code:`float16`, :code:`float32`, :code:`int32`}
15311531
:param input_precision: How to exercise the Tensor Cores for f32 x f32. If
@@ -1559,13 +1559,13 @@ def dot_scaled(lhs, lhs_scale, lhs_format, rhs, rhs_scale, rhs_format, acc=None,
15591559
:type lhs: 2D tensor representing fp4 or fp8 elements packed into uint8 for fp4 inputs, or in uint8 or the corresponding fp8 type for fp8 inputs.
15601560
:param lhs_scale: Scale factor for lhs tensor.
15611561
:type lhs_scale: e8m0 type represented as an uint8 tensor.
1562-
:param lhs_format: format of the lhs tensor. Available formats: {:code:`e2m1`, :code:`e4m3`, :code: `e5m2`}.
1562+
:param lhs_format: format of the lhs tensor. Available formats: {:code:`e2m1`, :code:`e4m3`, :code:`e5m2`}.
15631563
:type lhs_format: str
15641564
:param rhs: The second tensor to be multiplied.
15651565
:type rhs: 2D tensor representing fp8 or bf16 elements in uint8 or the corresponding fp8 type for fp8 inputs or bf16 for bf16 inputs.
15661566
:param rhs_scale: Scale factor for rhs tensor.
15671567
:type rhs_scale: e8m0 type represented as an uint8 tensor.
1568-
:param rhs_format: format of the rhs tensor. Available formats: {:code:`e4m3`, :code: `e5m2`, :code:`bf16`}.
1568+
:param rhs_format: format of the rhs tensor. Available formats: {:code:`e4m3`, :code:`e5m2`, :code:`bf16`}.
15691569
:type rhs_format: str
15701570
:param acc: The accumulator tensor. If not None, the result is added to this tensor.
15711571
"""

0 commit comments

Comments
 (0)