Skip to content

Commit

Permalink
err, this
Browse files Browse the repository at this point in the history
  • Loading branch information
geohot committed Aug 23, 2023
1 parent b4feaac commit f7ecc16
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions teenygrad/lazy.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,11 +4,6 @@
from teenygrad.ops import UnaryOps, BinaryOps, ReduceOps, TernaryOps, LoadOps
import numpy as np

class Device:
DEFAULT = "CPU"
_buffers = ["CPU"]
def canonicalize(x): return "CPU"

def shape_to_axis(old_shape:Tuple[int, ...], new_shape:Tuple[int, ...]) -> Tuple[int, ...]:
assert len(old_shape) == len(new_shape), "reduce shapes must have same dimensions"
return tuple(i for i,(a,b) in enumerate(zip(old_shape, new_shape)) if a != b)
Expand Down

0 comments on commit f7ecc16

Please sign in to comment.