Fix

Author: Binyang2014

include/mscclpp/port_channel.hpp

@@ -51,9 +51,9 @@ class ProxyService : public BaseProxyService {
   MemoryId addMemory(RegisteredMemory memory);
 
   /// Get the next available memory ID.
-  /// @param count The number of consecutive IDs required.
+  /// @param count The number of consecutive IDs required (default: 1).
   /// @return The first ID of an available range [first, first + count).
-  MemoryId nextMemoryId(uint32_t count) const;
+  MemoryId nextMemoryId(uint32_t count = 1) const;
 
   /// Get a semaphore by ID.
   /// @param id The ID of the semaphore.

src/executor/executor.cc

@@ -241,10 +241,10 @@ struct Executor::Impl {
         this->defaultScratchBuffer = GpuBuffer(this->defaultScratchBufferSize).memory();
       }
       if (scratchBufferSize > this->defaultScratchBufferSize) {
-        throw Error(
-            "DefaultScratchBuffer size not enough. Consider increasing the default scratch buffer size or disabling "
-            "resource reuse.",
-            ErrorCode::ExecutorError);
+        throw Error("Scratch buffer size (" + std::to_string(scratchBufferSize) +
+                        " bytes) exceeds default buffer size (" + std::to_string(this->defaultScratchBufferSize) +
+                        " bytes). Consider increasing the default scratch buffer size or disabling resource reuse.",
+                    ErrorCode::ExecutorError);
       }
       context.scratchBufferSize = this->defaultScratchBufferSize;
       context.scratchBuffer = this->defaultScratchBuffer;

src/port_channel.cc

@@ -63,6 +63,9 @@ MSCCLPP_API_CPP MemoryId ProxyService::addMemory(RegisteredMemory memory) {
 }
 
 MSCCLPP_API_CPP MemoryId ProxyService::nextMemoryId([[maybe_unused]] uint32_t count) const {
+  if (count == 0) {  
+    throw Error("count must be greater than 0", ErrorCode::InvalidUsage);
+  }  
   MemoryId firstId = memories_.size();
   return firstId;
 }