File tree Expand file tree Collapse file tree 11 files changed +14
-21
lines changed
Expand file tree Collapse file tree 11 files changed +14
-21
lines changed Original file line number Diff line number Diff line change 1313#include < mscclpp/gpu_data_types.hpp>
1414#include < mscclpp/memory_channel.hpp>
1515#include < mscclpp/memory_channel_device.hpp>
16- #include < mscclpp/nvls.hpp>
1716#include < mscclpp/packet_device.hpp>
17+ #include < mscclpp/switch_channel.hpp>
1818#include < type_traits>
1919
2020#if defined(ENABLE_NPKIT)
Original file line number Diff line number Diff line change 11#include < algorithm>
22#include < mscclpp/core.hpp>
33#include < mscclpp/memory_channel.hpp>
4- #include < mscclpp/nvls .hpp>
4+ #include < mscclpp/switch_channel .hpp>
55
66#include " common.hpp"
77
@@ -98,8 +98,8 @@ std::vector<mscclpp::SwitchChannel> setupNvlsChannels(std::vector<std::shared_pt
9898
9999 for (int idx = 0 ; idx < nSwitchChannels; ++idx) {
100100 std::shared_ptr<mscclpp::NvlsConnection> nvlsConnection = conns[idx];
101- mscclpp::SwitchChannel SwitchChannel = nvlsConnection->bindAllocatedMemory ((CUdeviceptr)buffer, bufferSize);
102- channels.push_back (SwitchChannel );
101+ mscclpp::SwitchChannel switchChannel = nvlsConnection->bindAllocatedMemory ((CUdeviceptr)buffer, bufferSize);
102+ channels.push_back (switchChannel );
103103 }
104104 return channels;
105105}
Original file line number Diff line number Diff line change 88#include < mscclpp/core.hpp>
99#include < mscclpp/env.hpp>
1010#include < mscclpp/memory_channel.hpp>
11- #include < mscclpp/nvls .hpp>
11+ #include < mscclpp/switch_channel .hpp>
1212#include < vector>
1313
1414#if defined(__HIP_PLATFORM_AMD__)
Original file line number Diff line number Diff line change 44#include < algorithm>
55#include < filesystem>
66#include < functional>
7- #include < mscclpp/concurrency_device.hpp>
87#include < mscclpp/core.hpp>
98#include < mscclpp/env.hpp>
109#include < mscclpp/executor.hpp>
11- #include < mscclpp/memory_channel.hpp>
12- #include < mscclpp/memory_channel_device.hpp>
13- #include < mscclpp/nvls.hpp>
1410#include < mscclpp/utils.hpp>
15- #include < queue>
16- #include < sstream>
1711#include < unordered_map>
1812#include < vector>
1913#if defined(ENABLE_NPKIT)
Original file line number Diff line number Diff line change 66
77#include < memory>
88#include < mscclpp/memory_channel.hpp>
9- #include < mscclpp/nvls.hpp>
109#include < mscclpp/port_channel.hpp>
10+ #include < mscclpp/switch_channel.hpp>
1111#include < vector>
1212
1313namespace mscclpp {
Original file line number Diff line number Diff line change 11// Copyright (c) Microsoft Corporation.
22// Licensed under the MIT license.
33
4- #ifndef MSCCLPP_NVLS_HPP_
5- #define MSCCLPP_NVLS_HPP_
4+ #ifndef MSCCLPP_SWITCH_CHANNEL_HPP_
5+ #define MSCCLPP_SWITCH_CHANNEL_HPP_
66
77#include < mscclpp/gpu_utils.hpp>
88#include < mscclpp/switch_channel_device.hpp>
@@ -68,4 +68,4 @@ std::shared_ptr<NvlsConnection> connectNvlsCollective(std::shared_ptr<Communicat
6868
6969} // namespace mscclpp
7070
71- #endif // MSCCLPP_NVLS_HPP_
71+ #endif // MSCCLPP_SWITCH_CHANNEL_HPP_
Original file line number Diff line number Diff line change 99#include < nanobind/stl/vector.h>
1010
1111#include < mscclpp/core.hpp>
12- #include < mscclpp/nvls .hpp>
12+ #include < mscclpp/switch_channel .hpp>
1313
1414namespace nb = nanobind;
1515using namespace mscclpp ;
Original file line number Diff line number Diff line change 33
44#include < mscclpp/executor.hpp>
55#include < mscclpp/memory_channel.hpp>
6- #include < mscclpp/nvls.hpp>
76#include < mscclpp/port_channel.hpp>
8- #include < set >
7+ #include < mscclpp/switch_channel.hpp >
98
109#include " debug.h"
1110#include " execution_kernel.hpp"
Original file line number Diff line number Diff line change 55#define MSCCLPP_EXECUTION_COMMON_HPP_
66
77#include < mscclpp/memory_channel.hpp>
8- #include < mscclpp/nvls.hpp>
98#include < mscclpp/port_channel.hpp>
9+ #include < mscclpp/switch_channel.hpp>
1010
1111namespace mscclpp {
1212
Original file line number Diff line number Diff line change 66
77#include < algorithm>
88#include < mscclpp/core.hpp>
9- #include < mscclpp/nvls .hpp>
9+ #include < mscclpp/switch_channel .hpp>
1010#include < mscclpp/utils.hpp>
1111
1212#include " api.h"
You can’t perform that action at this time.
0 commit comments