github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-10 14:45:35 -05:00
Code Issues Packages Projects Releases Wiki Activity

good changes from tensor_cores branch (#1005)

Browse Source 
* good changes from tensor_cores branch

* touchups

* real_strides fixup

* refactor merge_views
This commit is contained in:
George Hotz
2023-06-18 20:28:06 -07:00
committed by GitHub
parent ccb51ff5b0
commit 5428b5d774
6 changed files with 118 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tinygrad/runtime/ops_gpu.py
View File

@@ -28,7 +28,6 @@ class _CL:
for q in self.cl_queue: q.finish()
CL = _CL()
# TODO: merge CLImage in here
class CLBuffer(RawBufferCopyInOut):
def __init__(self, size, dtype, device='0'):
assert not OSX or dtype != dtypes.float64, "OpenCL on Mac doesn't support float64"