/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
H A D | Functional.cpp | 224 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 D | PyProcessGroup.hpp | 168 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 D | FakeProcessGroup.hpp | 136 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::FakeProcessGroup
|
H A D | Backend.hpp | 226 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::Backend
|
H A D | ProcessGroupGloo.hpp | 345 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced(
|
H A D | Ops.cpp | 336 ->reduce_scatter_tensor_coalesced( \
|
H A D | ProcessGroup.hpp | 397 virtual c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced( in reduce_scatter_tensor_coalesced() function in c10d::ProcessGroup
|
H A D | ProcessGroupNCCL.hpp | 605 c10::intrusive_ptr<Work> reduce_scatter_tensor_coalesced(
|
H A D | ProcessGroupGloo.cpp | 1949 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 D | init.cpp | 2025 &::c10d::ProcessGroup::reduce_scatter_tensor_coalesced, in c10d_init()
|
H A D | ProcessGroupNCCL.cpp | 4040 c10::intrusive_ptr<Work> ProcessGroupNCCL::reduce_scatter_tensor_coalesced( in checkForNCCLErrorsInternal() function in c10d::ProcessGroupNCCL
|
/aosp_15_r20/external/pytorch/test/distributed/ |
H A D | test_c10d_functional_native.py | 19 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 D | test_functional_api.py | 357 res = ft_c.reduce_scatter_tensor_coalesced(tensors, "sum", [0, 0], mesh)
|
H A D | test_inductor_collectives.py | 1088 tensor_list = torch.ops.c10d_functional.reduce_scatter_tensor_coalesced(
|
H A D | test_c10d_gloo.py | 740 work = dist.group.WORLD.reduce_scatter_tensor_coalesced(outputs, inputs)
|
/aosp_15_r20/external/pytorch/torch/distributed/ |
H A D | _functional_collectives_impl.py | 83 return torch.ops._c10d_functional.reduce_scatter_tensor_coalesced(
|
H A D | _functional_collectives.py | 389 def reduce_scatter_tensor_coalesced( function 423 …tensor_list = torch.ops._c10d_functional.reduce_scatter_tensor_coalesced( # type: ignore[attr-def…
|
H A D | distributed_c10d.py | 2308 work = group.reduce_scatter_tensor_coalesced(outputs, inputs, reduce_opts)
|
/aosp_15_r20/external/pytorch/torch/distributed/tensor/debug/ |
H A D | _comm_mode.py | 39 funcol_native.reduce_scatter_tensor_coalesced: funcol_py.reduce_scatter_tensor_coalesced,
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_snode_runtime.py | 310 rs = _c10d.reduce_scatter_tensor_coalesced(
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
H A D | multi_threaded_pg.py | 389 …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.pyi | 360 def reduce_scatter_tensor_coalesced(
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | lowering.py | 6398 @register_lowering(_c10d_functional.reduce_scatter_tensor_coalesced) 6403 _c10d_functional.reduce_scatter_tensor_coalesced.default,
|