hotfix: GRAPH_ONE_KERNEL + fix timing

This commit is contained in:
George Hotz
2025-02-06 17:52:20 +08:00
parent 1c53e8bf27
commit ae45826758
2 changed files with 3 additions and 2 deletions

View File

@@ -19,8 +19,8 @@ if __name__ == "__main__":
st = time.perf_counter()
c = plus(a,b)
et = time.perf_counter() - st
Device[c.device].synchronize()
print(f"nosync {i}: {et*1e6:.2f} us")
Device[c.device].synchronize()
for i in range(5):
st = time.perf_counter()
c = plus(a,b)