Skip to content

Commit d4cce9f

Browse files
authored
[FRONTEND] Fix missing handling for None acc in dot_scaled (#8658)
1 parent 62094e1 commit d4cce9f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/triton/language/core.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2096,6 +2096,7 @@ def dot_scaled(lhs, lhs_scale, lhs_format, rhs, rhs_scale, rhs_format, acc=None,
20962096
:type rhs_k_pack: bool, optional
20972097
"""
20982098
out_dtype = _unwrap_if_constexpr(out_dtype)
2099+
acc = _unwrap_if_constexpr(acc)
20992100
assert out_dtype == float32, "Only float32 is supported for out_dtype at the moment"
21002101
return _semantic.dot_scaled(lhs, lhs_scale, lhs_format, rhs, rhs_scale, rhs_format, acc, fast_math, lhs_k_pack,
21012102
rhs_k_pack, out_dtype)

0 commit comments

Comments
 (0)