mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

oops, broke mem estimates

Browse Source
This commit is contained in:
George Hotz
2023-01-28 20:21:31 -08:00
parent a500e79bd1
commit a9cabce791
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extra/kernel_search.py
View File

@@ -44,9 +44,9 @@ def apply_intervention(k, typ, *dat):
def search(ast):
# get baseline
k = CLASTKernel(ast)
CL.time_sum = 0
#k.hand_coded_optimizations()
k.codegen()(*k.bufs)
for i in range(3):
CL.time_sum = 0
k.codegen()(*k.bufs)
winning_interventions = []
best_time = baseline = CL.time_sum
@@ -68,7 +68,7 @@ def search(ast):
best_time = CL.time_sum
winning_interventions.append(inter)
for i in range(100):
for i in range(200):
try:
test()
except Exception as e:

2
tinygrad/llops/ops_gpu.py
View File

@@ -305,7 +305,7 @@ class CLASTKernel(ASTKernel):
# compile kernel
self.fxn = CLProgram(function_name, ' '.join(self.kernel), op_estimate=self.info.flops)
mem_estimate = sum(prod(x.shape) for x in self.sts)
mem_estimate = sum(prod(x._base_shape) for x in self.bufs)
if DEBUG >= 3 and len(self.bufs_to_delete): print(f"deleting buffers {self.bufs_to_delete}")
def runner(*bufs):