microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 01:10:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

doc string

Browse Source
This commit is contained in:
Binyang Li
2023-07-27 09:36:11 +00:00
parent 59e15c80de
commit b9ec5a6afb
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/mscclpp/proxy_channel.hpp
View File

@@ -41,6 +41,9 @@ class ProxyService : public BaseProxyService {
/// @return The ID of the semaphore.
SemaphoreId addSemaphore(std::shared_ptr<Connection> connection);
/// Add a pitch pair to the proxy service.
/// @param id The ID of the semaphore.
/// @param pitch The pitch pair.
void addPitch(SemaphoreId id, std::pair<uint64_t, uint64_t> pitch);
/// Register a memory region with the proxy service.

2
src/proxy_channel.cc
View File

@@ -67,7 +67,7 @@ ProxyHandlerResult ProxyService::handleTrigger(ProxyTrigger triggerRaw) {
RegisteredMemory& dst = memories_[trigger->fields.dstMemoryId];
RegisteredMemory& src = memories_[trigger->fields.srcMemoryId];
if (trigger->fields2D.multiDimensionFlag) {
std::pair<uint64_t, uint64_t>& pitch = pitches_[trigger->fields.chanId];
std::pair<uint64_t, uint64_t>& pitch = pitches_.at(trigger->fields.chanId);
semaphore->connection()->write2D(dst, trigger->fields.dstOffset, pitch.first, src, trigger->fields.srcOffset,
pitch.second, trigger->fields2D.width, trigger->fields2D.height);
} else {