Home
last modified time | relevance | path

Searched refs:reduce_scatter_tensor_coalesced (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DFunctional.cpp224 std::vector<at::Tensor> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function
240 auto work = group->reduce_scatter_tensor_coalesced(outputs, inputs, opts); in reduce_scatter_tensor_coalesced()
253 return reduce_scatter_tensor_coalesced( in reduce_scatter_tensor()
365 ::reduce_scatter_tensor_coalesced), in TORCH_LIBRARY()
H A DPyProcessGroup.hpp168 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::PyProcessGroup
175 reduce_scatter_tensor_coalesced, /* Name of function in C++ */ in reduce_scatter_tensor_coalesced()
H A DFakeProcessGroup.hpp136 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::FakeProcessGroup
H A DBackend.hpp226 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::Backend
H A DProcessGroupGloo.hpp345 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced(
H A DOps.cpp336 ->reduce_scatter_tensor_coalesced( \
H A DProcessGroup.hpp397 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::ProcessGroup
H A DProcessGroupNCCL.hpp605 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced(
H A DProcessGroupGloo.cpp1949 return reduce_scatter_tensor_coalesced(outputTensors, inputTensors, opts); in _reduce_scatter_base()
1952 c10::intrusive_ptr<Work> ProcessGroupGloo::reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::ProcessGroupGloo
H A Dinit.cpp2025 &::c10d::ProcessGroup::reduce_scatter_tensor_coalesced, in c10d_init()
H A DProcessGroupNCCL.cpp4040 c10::intrusive_ptr<Work> ProcessGroupNCCL::reduce_scatter_tensor_coalesced( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_functional_native.py19 reduce_scatter_tensor_coalesced,
325 outputs = torch.ops._c10d_functional.reduce_scatter_tensor_coalesced(
336 outputs = reduce_scatter_tensor_coalesced(
737 rs0 = funcol.reduce_scatter_tensor_coalesced(
H A Dtest_functional_api.py357 res = ft_c.reduce_scatter_tensor_coalesced(tensors, "sum", [0, 0], mesh)
H A Dtest_inductor_collectives.py1088 tensor_list = torch.ops.c10d_functional.reduce_scatter_tensor_coalesced(
H A Dtest_c10d_gloo.py740 work = dist.group.WORLD.reduce_scatter_tensor_coalesced(outputs, inputs)
/aosp_15_r20/external/pytorch/torch/distributed/
H A D_functional_collectives_impl.py83 return torch.ops._c10d_functional.reduce_scatter_tensor_coalesced(
H A D_functional_collectives.py389 def reduce_scatter_tensor_coalesced( function
423 …tensor_list = torch.ops._c10d_functional.reduce_scatter_tensor_coalesced( # type: ignore[attr-def…
H A Ddistributed_c10d.py2308 work = group.reduce_scatter_tensor_coalesced(outputs, inputs, reduce_opts)
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/
H A D_comm_mode.py39 funcol_native.reduce_scatter_tensor_coalesced: funcol_py.reduce_scatter_tensor_coalesced,
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_snode_runtime.py310 rs = _c10d.reduce_scatter_tensor_coalesced(
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py389 …def reduce_scatter_tensor_coalesced(self, output_tensors, input_tensors, opts=ReduceScatterOptions… member in ProcessLocalGroup
/aosp_15_r20/external/pytorch/torch/_C/
H A D_distributed_c10d.pyi360 def reduce_scatter_tensor_coalesced(
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py6398 @register_lowering(_c10d_functional.reduce_scatter_tensor_coalesced)
6403 _c10d_functional.reduce_scatter_tensor_coalesced.default,