Home
last modified time | relevance | path

Searched refs:ModularIndexing (Results 1 – 14 of 14) sorted by relevance

/external/pytorch/test/inductor/
Dtest_indexing.py23 ModularIndexing,
43 + ModularIndexing(i1, 1, 64)
44 + 64 * ModularIndexing(i1 + 64 * r3, 64, 2)
50 i1 + 128 * i2 + 64 * ModularIndexing(r3, 1, 2),
58 expr = ModularIndexing(i1 - 15, 1, 64)
61 ModularIndexing(i1 - 15, 1, 64),
69 expr = ModularIndexing(2 * i2 + r3, 1, 64)
75 self.assertEqual(ModularIndexing(r3 * i0, r3, 10), ModularIndexing(i0, 1, 10))
79 ModularIndexing(i0 + i1 * 10, 1, 10), ModularIndexing(i0, 1, 10)
84 ModularIndexing(i0 + i1 * 20, 2, 10), ModularIndexing(i0, 2, 10)
[all …]
Dtest_loop_ordering.py23 from torch.utils._sympy.functions import ModularIndexing
138 "primal", i3 + 49 * i2 + 2401 * ModularIndexing(i0, 1, 64)
160 z3 + 49 * z2 + 2401 * ModularIndexing(z0, 1, 64),
165 z2 + 49 * z1 + 2401 * ModularIndexing(z3, 1, 64),
/external/pytorch/torch/_inductor/
Dsizevars.py23 from torch.utils._sympy.functions import FloorDiv, ModularIndexing
209 return ModularIndexing(base, divisor, modulus)
211 if expr.has(ModularIndexing):
213 ModularIndexing(
694 if not isinstance(x, ModularIndexing):
705 if isinstance(index, ModularIndexing):
719 return ModularIndexing(x2, 1, mod)
806 if not isinstance(expr, sympy.Add) or not expr.has(ModularIndexing):
831 m1 = term1.match(scale * ModularIndexing(base, divisor, mod1))
837 * ModularIndexing(m1[base], m1[divisor] * m1[mod1], mod2)
[all …]
Dindex_propagation.py32 from torch.utils._sympy.functions import FloorDiv, ModularIndexing, Where
138 result_expr = ModularIndexing(x.expr, sympy.Integer(1), y.expr)
155 result_expr = ModularIndexing(x.expr, sympy.Integer(1), y.expr)
Dutils.py73 ModularIndexing,
640 if isinstance(expr, (ModularIndexing, CleanDiv, FloorDiv, Identity)):
Dlowering.py41 ModularIndexing,
998 index[i] = ModularIndexing(index[i], 1, old_size[i])
2564 ModularIndexing(idx[dim] - start, 1, step), torch.int64
Dir.py60 from torch.utils._sympy.functions import CleanDiv, FloorDiv, ModularIndexing
2258 view_expr.append(ModularIndexing(var, divisor, modulus))
2262 view_expr.append(ModularIndexing(var, divisor, modulus))
/external/pytorch/torch/utils/_sympy/
Dinterp.py33 ModularIndexing,
98 ModularIndexing: "modular_indexing",
Dfunctions.py256 class ModularIndexing(sympy.Function): class
280 return ModularIndexing(
308 return ModularIndexing(sum(new_terms), divisor, modulus)
311 return ModularIndexing(base.args[0], base.args[1] * divisor, modulus)
/external/pytorch/torch/_inductor/codegen/
Dhalide.py29 from ...utils._sympy.functions import FloorDiv, ModularIndexing
766 if index.has(ModularIndexing):
768 ModularIndexing(
871 ModularIndexing(full_index, node.divisor, node.length),
Dsimd.py31 from torch.utils._sympy.functions import FloorDiv, Identity, ModularIndexing
171 expr = ModularIndexing(self.index_sym(), divisor, length)
274 assert isinstance(self.expr, (FloorDiv, ModularIndexing)), type(self.expr)
442 if not isinstance(index, ModularIndexing):
Dcpp.py20 from torch.utils._sympy.functions import CeilDiv, FloorDiv, ModularIndexing
291 result = ModularIndexing(var, divisor, modulus)
307 if index.has(ModularIndexing):
308 index = index.replace(ModularIndexing(var, div, mod), visit_modular_indexing)
4179 expr.has(ModularIndexing) for expr in node._body.indexing_exprs.values()
Dtriton.py32 from torch.utils._sympy.functions import CeilDiv, FloorDiv, ModularIndexing
1419 (index.count(FloorDiv) + index.count(ModularIndexing)),
1447 ModularIndexing(index_var, numel, dim)
/external/pytorch/torch/_export/serde/
Dserialize.py1866 "ModularIndexing": torch.utils._sympy.functions.ModularIndexing,