├── README.md ├── custom_backend ├── .gitignore ├── README.md ├── example.py ├── include │ └── dummy.hpp ├── setup.py └── src │ └── dummy.cpp └── custom_process_group ├── .gitignore ├── README.md ├── example.py ├── include └── dummy.hpp ├── setup.py └── src └── dummy.cpp /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/README.md -------------------------------------------------------------------------------- /custom_backend/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/.gitignore -------------------------------------------------------------------------------- /custom_backend/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/README.md -------------------------------------------------------------------------------- /custom_backend/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/example.py -------------------------------------------------------------------------------- /custom_backend/include/dummy.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/include/dummy.hpp -------------------------------------------------------------------------------- /custom_backend/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/setup.py -------------------------------------------------------------------------------- /custom_backend/src/dummy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_backend/src/dummy.cpp -------------------------------------------------------------------------------- /custom_process_group/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/.gitignore -------------------------------------------------------------------------------- /custom_process_group/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/README.md -------------------------------------------------------------------------------- /custom_process_group/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/example.py -------------------------------------------------------------------------------- /custom_process_group/include/dummy.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/include/dummy.hpp -------------------------------------------------------------------------------- /custom_process_group/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/setup.py -------------------------------------------------------------------------------- /custom_process_group/src/dummy.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/H-Huang/torch_collective_extension/HEAD/custom_process_group/src/dummy.cpp --------------------------------------------------------------------------------