Searched refs:DeferredLine (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/_inductor/codegen/ |
| D | triton.py | 59 DeferredLine,
1635 ) -> Tuple[str, Optional[DeferredLine], str]:
1653 DeferredLine(
1657 advance_block_ptr = DeferredLine(
1834 self.stores.writeline(DeferredLine(name, "tl.debug_barrier()"))
1851 self.stores.writeline(DeferredLine(name, line))
2138 DeferredLine(
2152 DeferredLine(
|
| D | triton_combo_kernel.py | 31 DeferredLine,
1031 elif isinstance(line, DeferredLine) and (
1039 new_line = DeferredLine(line.name, modified_line)
|
| D | cpp.py | 55 DeferredLine,
1918 self.stores.writeline(DeferredLine(name, line))
1951 DeferredLine(name, f"{var}[{cexpr_index(index)}] = {value};")
2517 self.stores.splice(code.map(lambda x: DeferredLine(name, x)))
2527 self.stores.splice(code.map(lambda x: DeferredLine(name, x)))
2535 self.stores.writeline(DeferredLine(name, line))
2740 self.reduction_suffix.splice(code.map(lambda x: DeferredLine(name, x)))
3100 self.poststores.writeline(DeferredLine(name, load_or_store))
3148 self.stores.writeline(DeferredLine(name, line))
|
| D | halide.py | 49 DeferredLine,
1162 DeferredLine(name, f"{var}[{undef_dims}] = hl.undef({var}.type())")
1175 self.body.writeline(DeferredLine(name, line))
|
| D | wrapper.py | 57 from .common import CodeGen, DeferredLine, IndentedBuffer, PythonPrinter
1846 DeferredLine(
|
| D | common.py | 1143 class DeferredLine(DeferredLineBase): class
1165 return DeferredLine(self.name, line)
|