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

fix KOPT

Browse Source
This commit is contained in:
George Hotz
2023-02-07 06:56:12 -06:00
parent 7604b17fbf
commit d93563f39f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extra/kernel_search.py
View File

@@ -17,6 +17,7 @@ intervention_cache = dbm.open('/tmp/kopt.db', 'c')
Interventions = Enum("Interventions", ["SWAP", "UPCAST", "SHIFT", "REDUCE"])
def get_interventions(k):
k.process()
p1, p2, p3, p4, p5 = [], [], [], [], []
p1 = [(Interventions.SWAP, x) for x in itertools.combinations(range(k.first_reduce), 2)]
p2 = [(Interventions.SWAP, x) for x in itertools.combinations(range(k.first_reduce, k.shape_len), 2)]