Home
last modified time | relevance | path

Searched refs:triton_kernel_wrapper_functional (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtriton_kernel_wrap.py555 triton_kernel_wrapper_functional = TritonKernelWrapperFunctional() variable
642 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_impl
725 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 …]
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dreinplace.py12 triton_kernel_wrapper_functional,
359 inplaceable_triton_ops = {triton_kernel_wrapper_functional}
409 src.args[0].target == triton_kernel_wrapper_functional
H A Dpost_grad.py833 CallFunctionVarArgs(torch.ops.higher_order.triton_kernel_wrapper_functional),
890 target=torch.ops.higher_order.triton_kernel_wrapper_functional,
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_triton_kernels.py12 triton_kernel_wrapper_functional,
101 out_dict = triton_kernel_wrapper_functional(
129 out = triton_kernel_wrapper_functional(
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dpartitioners.py828 if b.target != torch.ops.higher_order.triton_kernel_wrapper_functional: