Home
last modified time | relevance | path

Searched refs:new_define_kernel (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py1786 def new_define_kernel(wrapper, name, kernel_code, metadata, *args, **kwargs): function
1791 with unittest.mock.patch.object(WrapperCodeGen, "define_kernel", new_define_kernel):