use unittest helpers in TestTensorMetadata [pr] (#7329)

* use unittest helpers in TestTensorMetadata [pr]

* fix that

* 5 args
This commit is contained in:
qazal
2024-10-28 12:38:30 +02:00
committed by GitHub
parent 96fcc47e27
commit e46edc22aa
3 changed files with 26 additions and 26 deletions

View File

@@ -29,7 +29,7 @@ if REF == "master": SKIP_PROCESS_REPLAY = True
# *** recreators
def recreate_sched(sink:UOp, ctx:ScheduleItemContext) -> UOp: return full_ast_rewrite(sink, ctx)
def recreate_kernel(ast:UOp, opts:Renderer, applied_opts:List[Opt], name:str) -> str:
def recreate_kernel(ast:UOp, opts:Renderer, applied_opts:List[Opt], name:str, _) -> str:
k = Kernel(ast, opts=opts)
for opt in applied_opts: k.apply_opt(opt)
# NOTE: replay with the captured renderer, not the one in master