commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fngrl

This commit is contained in:
Yassine Yousfi 2022-10-06 20:03:59 -07:00
parent 2993dfe921
commit c86edb947f
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
openpilot/compile.py
View File

@ -34,7 +34,7 @@ def get_random_input_tensors():
"big_input_imgs": np.random.randn(*(1, 12, 128, 256))*256,
"desire": np.zeros((1,100, 8)),
"traffic_convention": np.array([[1., 0.]]),
"features_buffer": np.random.randn(*(1, 99, 2048))
"features_buffer": np.random.randn(*(1, 99, 128))
#"initial_state": np.zeros((1, 768))
}
@ -108,7 +108,7 @@ def compile(input, output_fn):
args[1] = [min(MAX_WORKGROUP, args[0][0]), 1]
try:
e = prg.clprg(CL().cl_queue, *args)
except cl.LogicError:
except (cl.LogicError, cl.RuntimeError):
# INVALID_WORK_GROUP_SIZE
args[1] = None
continue
@ -136,7 +136,7 @@ def compile(input, output_fn):
if bad: continue
try:
e = prg.clprg(CL().cl_queue, *args)
except cl.LogicError:
except (cl.LogicError, cl.RuntimeError):
# INVALID_WORK_GROUP_SIZE
continue
CL().cl_queue.finish()