github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 13:28:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove unused CPUBuffer functions

Browse Source
This commit is contained in:
George Hotz
2022-06-09 09:45:39 -07:00
parent 9d85067ad8
commit 273af8d732
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tinygrad/llops/ops_cpu.py
View File

@@ -2,19 +2,17 @@ import numpy as np
from tinygrad.helpers import UnaryOps, BinaryOps, ReduceOps
class CPUBuffer(np.ndarray):
def toCPU(x): return x
def log(x): return np.log(x)
def exp(x): return np.exp(x)
def sign(x): return np.sign(x)
def relu(x): return np.maximum(x, 0)
def type(x, tt): return x.astype(tt)
def custompad(x, padding): return np.pad(x, padding)
def permute(x, order): return x.transpose(order)
def expand(x, shp): return np.broadcast_to(x, shp)
def exp(x): return np.exp(x)
def log(x): return np.log(x)
def sign(x): return np.sign(x)
def amax(x, *args, **kwargs): return np.amax(x, *args, **kwargs)
def permute(x, order): return x.transpose(order)
def custompad(x, padding): return np.pad(x, padding)
@staticmethod
def fromCPU(x): return x
def toCPU(x): return x
def unary_op(op, x, ret):
if op == UnaryOps.RELU: ret[:] = x.relu()