mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
all ops supported on GPU now
This commit is contained in:
@@ -80,7 +80,7 @@ optim.step()
|
|||||||
|
|
||||||
### GPU Support?!
|
### GPU Support?!
|
||||||
|
|
||||||
tinygrad supports GPUs through PyOpenCL. Not all ops are supported yet on the backward pass.
|
tinygrad supports GPUs through PyOpenCL.
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from tinygrad.tensor import Tensor
|
from tinygrad.tensor import Tensor
|
||||||
@@ -89,7 +89,7 @@ from tinygrad.tensor import Tensor
|
|||||||
|
|
||||||
### ANE Support?!?!
|
### ANE Support?!?!
|
||||||
|
|
||||||
So it doesn't work yet, but see the ane directory for code to use the Apple Neural Engine at a low level.
|
So it doesn't work yet, but see the `ane` directory for code to use the Apple Neural Engine at a low level.
|
||||||
|
|
||||||
### ImageNet inference
|
### ImageNet inference
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user