(self)
| 41 | def shape_str_to_axis(self, nms:list[str]) -> tuple[int, ...]: return tuple([self.shape_str().index(x) for x in nms]) |
| 42 | |
| 43 | def copy(self) -> Scheduler: |
| 44 | ret = Scheduler(self.ast, self.ren) |
| 45 | ret.dont_use_locals = self.dont_use_locals |
| 46 | ret.applied_opts = self.applied_opts[:] |
| 47 | if hasattr(self, 'tensor_core'): ret.tensor_core = self.tensor_core |
| 48 | return ret |
| 49 | |
| 50 | kernel_cnt: Final[defaultdict[str, int]] = defaultdict(int) |
| 51 | def get_optimized_ast(self, name_override:str|None=None) -> UOp: |
no test coverage detected