github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

tests from new lazy branch (#2774)

Browse Source 
* tests from new lazy branch

* fix lin 11

* that was needed

* doesn't fail

* mark

* meant that

* llvm passes
This commit is contained in:
George Hotz
2023-12-14 23:06:39 -08:00
committed by GitHub
parent a044125c39
commit c6eb618013
8 changed files with 64 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
openpilot/compile2.py
View File

@@ -13,7 +13,6 @@ import onnx
from tqdm import tqdm
from typing import Tuple, List, Optional, Dict
from extra.onnx import get_run_onnx
from tinygrad.graph import log_schedule_item
from tinygrad import Tensor, Device
from tinygrad.helpers import dtypes, partition, GlobalCounters, Context, fetch, getenv, ImageDType, GRAPH, DEBUG
from tinygrad.realize import run_schedule, lower_schedule_item
@@ -111,10 +110,6 @@ if __name__ == "__main__":
image_count = sum(isinstance(si.out.dtype, ImageDType) for si in schedule)
print(f"**** running real kernels {image_count}/{len(schedule)} images ****")
if GRAPH:
for si in schedule_input: log_schedule_item(si)
for si in schedule: log_schedule_item(si)
GlobalCounters.reset()
run_schedule(schedule[:])