Searched refs:lib_impl (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/ |
| D | _functional_collectives.py | 936 lib_impl = torch.library.Library("_c10d_functional", "IMPL") variable
937 lib_impl.impl("all_reduce", _all_reduce_meta, "Meta")
938 lib_impl.impl("all_reduce_", _all_reduce__meta, "Meta")
939 lib_impl.impl("all_reduce_coalesced", _all_reduce_coalesced_meta, "Meta")
940 lib_impl.impl("all_reduce_coalesced_", _all_reduce_coalesced__meta, "Meta")
941 lib_impl.impl("wait_tensor", _wait_tensor_meta, "Meta")
942 lib_impl.impl(
945 lib_impl.impl("all_gather_into_tensor", _all_gather_into_tensor_native_meta, "Meta")
946 lib_impl.impl(
951 lib_impl.impl("reduce_scatter_tensor", _reduce_scatter_tensor_native_meta, "Meta")
[all …]
|