[c10d] Use macro to deduplicate codes (#101243) · sands-lab/NeuronaBox-PyTorch@4dbab17 · GitHub

Commit

[c10d] Use macro to deduplicate codes (pytorch#101243)

Browse files

Ops.cpp copies code for each of the three device keys (CPU, CUDA PrivateUse1).
Use macro to deduplicate it.
No logic change.
Cc @kumpera @H-Huang
Pull Request resolved: pytorch#101243
Approved by: https://github.com/H-Huang

Loading branch information

kwen2501 authored and pytorchmergebot committed May 12, 2023

1 parent 0be53d8 commit 4dbab17