Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/codegen/
Dmulti_kernel.py77 class MultiKernelState: class
Dwrapper.py36 from torch._inductor.codegen.multi_kernel import MultiKernelState
533 self.multi_kernel_state = MultiKernelState()