Searched refs:triton_kernel_wrapper_functional (Results 1 – 5 of 5) sorted by relevance
555 triton_kernel_wrapper_functional = TritonKernelWrapperFunctional() variable642 unwrapped_outputs = triton_kernel_wrapper_functional(665 @triton_kernel_wrapper_functional.py_impl(DispatchKey.CompositeExplicitAutograd)686 @triton_kernel_wrapper_functional.py_impl(FakeTensorMode)702 @triton_kernel_wrapper_functional.py_impl(ProxyTorchDispatchMode)708 triton_kernel_wrapper_functional,719 @triton_kernel_wrapper_functional.py_functionalize_impl725 outputs = triton_kernel_wrapper_functional(744 triton_kernel_wrapper_functional.fallthrough(DispatchKey.PythonDispatcher) # type: ignore[attr-def…745 triton_kernel_wrapper_functional.fallthrough(DispatchKey.PythonTLSSnapshot) # type: ignore[attr-de…[all …]
12 triton_kernel_wrapper_functional,359 inplaceable_triton_ops = {triton_kernel_wrapper_functional}409 src.args[0].target == triton_kernel_wrapper_functional
833 CallFunctionVarArgs(torch.ops.higher_order.triton_kernel_wrapper_functional),890 target=torch.ops.higher_order.triton_kernel_wrapper_functional,
12 triton_kernel_wrapper_functional,101 out_dict = triton_kernel_wrapper_functional(129 out = triton_kernel_wrapper_functional(
828 if b.target != torch.ops.higher_order.triton_kernel_wrapper_functional: