mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-14 08:57:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

rename llops to opencl

Browse Source
This commit is contained in:
George Hotz
2022-06-06 07:49:36 -07:00
parent 233c71a7ba
commit 30f55eaaba
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
tinygrad/llops/gpu.py → tinygrad/llops/opencl.py
View File

6
tinygrad/ops/ops_gpu.py
View File

@@ -2,9 +2,9 @@ import pyopencl as cl
import numpy as np
from tinygrad.helpers import binary_broadcast
from ..tensor import Function
from ..llops.gpu import GPUBuffer
from ..llops.gpu import unary_op, binary_op, reduce_op, perm_axis, inner_slice
from ..llops.gpu import matmul, conv, convdw, convdx
from ..llops.opencl import GPUBuffer
from ..llops.opencl import unary_op, binary_op, reduce_op, perm_axis, inner_slice
from ..llops.opencl import matmul, conv, convdw, convdx
def buffer_new(shape, zero=False):
return GPUBuffer(shape, hostbuf=None if not zero else np.zeros(shape, dtype=np.float32))