tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 11:56:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

see cpu speed again

Browse Source
This commit is contained in:
zibokapi
2025-06-24 21:11:05 +08:00
parent e2c9088db1
commit 1e366f2256
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/external/external_test_onnx_backend.py
Vendored
+1 -1
View File
@@ -30,7 +30,7 @@ class TinygradBackend(Backend):
input_initializer = [x.name for x in model.graph.initializer]
net_feed_input = [x for x in input_all if x not in input_initializer]
print("prepare", cls, device, net_feed_input)
model = Tensor(model.SerializeToString(), device="PYTHON")
model = Tensor(model.SerializeToString(), device="cpu")
run_onnx = OnnxRunner(onnx_load(model))
return TinygradModel(run_onnx, net_feed_input)