Home
last modified time | relevance | path

Searched refs:statically_known_multiple_of (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dtriton_utils.py123 offset_aligned = V.graph.sizevars.statically_known_multiple_of(
138 … return V.graph.sizevars.statically_known_multiple_of(x.expr, alignment) # type: ignore[arg-type]
140 …return V.graph.sizevars.statically_known_multiple_of(x.nbytes, alignment) # type: ignore[arg-type]
Dtriton.py308 and not sizevars.statically_known_multiple_of(
311 and not sizevars.statically_known_multiple_of(
1486 if not sizevars.statically_known_multiple_of(
1503 not sizevars.statically_known_multiple_of(numel, max_block)
2922 return V.graph.sizevars.statically_known_multiple_of(tree.numel, max_block)
Dsimd.py535 if not sv.statically_known_multiple_of(remaining[i], expr):
567 if not sv.statically_known_multiple_of(
1723 if V.graph.sizevars.statically_known_multiple_of(a1, b1):
/external/pytorch/torch/_inductor/
Dsizevars.py390 def statically_known_multiple_of( member in SizeVarAllocator
Dir.py2917 … return V.graph.sizevars.statically_known_multiple_of(offset, ALIGNMENT) # type: ignore[arg-type]