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] for loop negative step hotfix

Browse Source
This commit is contained in:
Phil Tillet
2023-03-01 23:45:03 -08:00
parent fa0fbc937f
commit 2660c814c9
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/triton/compiler.py
View File

@@ -742,6 +742,7 @@ class CodeGenerator(ast.NodeVisitor):
iv = self.builder.create_int_cast(iv, iv_ir_type, True)
if negative_step:
ub_si = self.builder.create_index_to_si(ub)
ub_si = self.builder.create_int_cast(ub_si, iv_ir_type, True)
iv = self.builder.create_sub(ub_si, iv)
self.lscope[node.target.id].handle.replace_all_uses_with(iv)
self.set_value(node.target.id, triton.language.core.tensor(iv, iv_type))