Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.

Commit a36aaed

Browse files
authored
Always do dense_indexing on atomic_add (#950)
1 parent a2b8a31 commit a36aaed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchinductor/codegen/triton.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,7 @@ def load(self, name: str, index: sympy.Expr, upcast: bool = False):
717717

718718
def store(self, name, index, value, mode=None):
719719
var = self.args.output(name)
720-
index, mask = self.indexing(index, value)
720+
index, mask = self.indexing(index, value, dense_indexing=True)
721721
if mode is None:
722722
line = f"tl.store({var} + {index}, {value}, {mask})"
723723
elif mode == "atomic_add":

0 commit comments

Comments
 (0)