KOPT is over, BEAM is upstream (#2071)

* create cache for q learning

* make linter happy

* global beam

* where it belongs

* bugfix

* ditch the kopt, use the beam

* faster lin and DEBUG=2 okay

* remove kopt, move search to features
This commit is contained in:
George Hotz
2023-10-16 09:46:03 -07:00
committed by GitHub
parent e4660b024f
commit c36d306606
14 changed files with 44 additions and 153 deletions

View File

@@ -13,9 +13,6 @@ OSX = platform.system() == "Darwin"
def compile_and_test_ast(ast, local_size=None):
k = CLCodegen(ast)
if getenv("KOPT", 0):
from extra.kernel_search import apply_optimization
apply_optimization(k, ast, 10, getenv("KCACHE", 0))
prg = k.codegen().build(CLProgram)
if local_size is not None: prg.local_size = local_size
for i in range(5): prg(prg.lower(k.bufs))