Searched refs:MultiKernelState (Results 1 – 2 of 2) sorted by relevance
77 class MultiKernelState: class
36 from torch._inductor.codegen.multi_kernel import MultiKernelState533 self.multi_kernel_state = MultiKernelState()