Searched refs:triton_kernel_wrapper_mutation (Results 1 – 4 of 4) sorted by relevance
537 triton_kernel_wrapper_mutation = TritonKernelWrapperMutation() variable558 @triton_kernel_wrapper_mutation.py_impl(DispatchKey.CompositeExplicitAutograd)580 @triton_kernel_wrapper_mutation.py_impl(FakeTensorMode)588 @triton_kernel_wrapper_mutation.py_impl(DispatchKey.Meta)609 @triton_kernel_wrapper_mutation.py_impl(ProxyTorchDispatchMode)615 triton_kernel_wrapper_mutation,627 @triton_kernel_wrapper_mutation.py_functionalize_impl677 triton_kernel_wrapper_mutation(735 triton_kernel_wrapper_mutation.fallthrough(DispatchKey.PythonDispatcher) # type: ignore[attr-defin…736 triton_kernel_wrapper_mutation.fallthrough(DispatchKey.PythonTLSSnapshot) # type: ignore[attr-defi…[all …]
965 triton_kernel_wrapper_mutation,986 triton_kernel_wrapper_mutation,
13 triton_kernel_wrapper_mutation,85 triton_kernel_wrapper_mutation(201 triton_kernel_wrapper_mutation(222 triton_kernel_wrapper_mutation(
21 from torch._higher_order_ops.triton_kernel_wrap import triton_kernel_wrapper_mutation6212 @register_lowering(triton_kernel_wrapper_mutation)