diff --git a/examples/llm.c/export.py b/examples/llm.c/export.py index a409ccca38..06e82e263b 100755 --- a/examples/llm.c/export.py +++ b/examples/llm.c/export.py @@ -8,7 +8,6 @@ from tinygrad.helpers import dedup, to_function_name, flatten, getenv, GRAPH, Gl from tinygrad.engine.schedule import create_schedule from tinygrad.engine.realize import memory_planner, run_schedule from tinygrad.ops import BufferOps, LoadOps -from tinygrad.runtime.ops_clang import CLANG_PROGRAM_HEADER TIMING = getenv("TIMING") @@ -48,7 +47,7 @@ if __name__ == "__main__": for ast in ast_dedup: k = Device["CLANG"].get_linearizer(*ast) k.linearize() - src = Device["CLANG"].compiler.render(to_function_name(k.name), k.uops).strip(CLANG_PROGRAM_HEADER) + src = Device["CLANG"].compiler.render(to_function_name(k.name), k.uops) srcs[ast] = (k.name, src) print("functions:", len(srcs)) used_buffers = dedup(flatten([si.bufs for si in sched])) @@ -70,10 +69,8 @@ if __name__ == "__main__": state_dict["adam_v_"+nm] = v named_buffers = {v.lazydata.base.buffer:k.replace(".", "_") for k,v in state_dict.items()} - if TIMING: - c_code = [CLANG_PROGRAM_HEADER, "#include ", "#include ", "#include "] - else: - c_code = ["#include ", CLANG_PROGRAM_HEADER] + c_code = ["#include ", "#include ", "#include "] + if TIMING: c_code += ["#include ", "#include "] c_code += [x[1].replace(" restrict ", " ")+"\n" for x in srcs.values()] premain = ["int main() {"]