From fd8e14c07a80b8ce18bcadaa3caeced9fe02e358 Mon Sep 17 00:00:00 2001 From: Roelof van Dijk <3604013+roelofvandijk@users.noreply.github.com> Date: Mon, 4 Sep 2023 20:39:50 +0200 Subject: [PATCH] fix: unused function (#1759) Co-authored-by: Roelof van Dijk --- tinygrad/codegen/linearizer.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tinygrad/codegen/linearizer.py b/tinygrad/codegen/linearizer.py index 697f3e903d..3e06fbb50a 100644 --- a/tinygrad/codegen/linearizer.py +++ b/tinygrad/codegen/linearizer.py @@ -1,5 +1,5 @@ from __future__ import annotations -from typing import List, Tuple, Any, Optional, cast, DefaultDict, NamedTuple, Dict, Iterator, Union, Sequence, Final, Set +from typing import List, Tuple, Any, Optional, cast, DefaultDict, NamedTuple, Dict, Union, Sequence, Final, Set import itertools, math, functools from collections import defaultdict from enum import Enum, auto @@ -44,10 +44,6 @@ def to_image_idx(base_shape:Tuple[int, ...], idxy:Node, valid:Node, validhacks=F if DEBUG >= 5: print("to_image_idx", base_shape, idx.min, idx.max, idy.min, idy.max, idx, idy) return idx, idy -def expand_idxs(idxs:Sequence[Node]) -> Iterator[Tuple[Node, ...]]: - for x in itertools.product(*[idx.expand() for idx in idxs[::-1]]): - yield x[::-1] - class UOp(NamedTuple): uop: UOps dtype: Optional[DType]