deepcrayon
/
tinygrab
1
0
Fork 0
Code

fix graph in openpilot/compile.py

pull/409/head
George Hotz 2022-10-28 08:55:34 -07:00
parent 1013540370
commit e6b65f8e01
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
openpilot/compile.py
View File

@ -12,7 +12,7 @@ DEBUGCL = int(os.getenv("DEBUGCL", 0))
import onnx
import numpy as np
import tinygrad.lazy as lazy
import tinygrad.graph as graph
from tinygrad.llops.ops_gpu import CL
from extra.utils import fetch
@ -55,8 +55,8 @@ def get_random_input_tensors(input_shapes):
def compile(dat, output_fn):
Tensor.no_grad = True
using_graph = lazy.GRAPH
lazy.GRAPH = False
using_graph = graph.GRAPH
graph.GRAPH = False
onnx_model = onnx.load(io.BytesIO(dat))
run_onnx = get_run_onnx(onnx_model)
@ -91,10 +91,10 @@ def compile(dat, output_fn):
# note, since CL.CACHE is enabled, it doesn't actually run the kernels
CL.CACHE = []
if using_graph: lazy.GRAPH = True
if using_graph: graph.GRAPH = True
CL.kernel_count = -1
tinygrad_out.realize()
lazy.GRAPH = False
graph.GRAPH = False
print("kernel count:", len(CL.CACHE))
assert len(CL.CACHE) <= ALLOWED_KERNEL_COUNT or ALLOWED_KERNEL_COUNT == 0, "too many kernels!"