Searched refs:TraceableTritonKernelWrapper (Results 1 – 2 of 2) sorted by relevance
225 from torch._higher_order_ops.triton_kernel_wrap import TraceableTritonKernelWrapper233 return TraceableTritonKernelWrapper(triton_kernel, None, None)
1016 class TraceableTritonKernelWrapper: class