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

[FRONTEND] add missing implicit constexpr conversion in dot (#2427)

Browse Source
This commit is contained in:
Philippe Tillet
2023-10-01 16:07:50 -07:00
committed by GitHub
parent 97e35b677b
commit a0025cfc44
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1002,6 +1002,7 @@ def dot(input, other, acc=None, allow_tf32=True, max_num_imprecise_acc=None, out
"""
allow_tf32 = _constexpr_to_value(allow_tf32)
out_dtype = _constexpr_to_value(out_dtype)
max_num_imprecise_acc = _constexpr_to_value(max_num_imprecise_acc)
return semantic.dot(input, other, acc, allow_tf32, max_num_imprecise_acc, out_dtype, _builder)