mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-10 22:54:59 -05:00
21 lines
600 B
Python
21 lines
600 B
Python
import numpy as np
|
|
|
|
from tinygrad.runtime.ops_gpu import CLCodegen
|
|
from tinygrad.codegen.assembly import AssemblyCodegen
|
|
|
|
from tinygrad.helpers import LazyNumpyArray, dtypes
|
|
from tinygrad.ops import LazyOp, BinaryOps
|
|
from tinygrad.lazy import LazyBuffer
|
|
from tinygrad.shape.shapetracker import ShapeTracker
|
|
|
|
ones = LazyNumpyArray.from_np(np.ones((3,), np.float32))
|
|
|
|
#target = "GPU"
|
|
target = "RDNA"
|
|
|
|
b1 = LazyBuffer.fromCPU(ones, target)
|
|
b2 = LazyBuffer.fromCPU(ones, target)
|
|
|
|
out = LazyBuffer(target, ShapeTracker((3,)), BinaryOps, LazyOp(BinaryOps.ADD, (b1, b2)), dtypes.float32)
|
|
print(out.toCPU())
|