TODO: capture Status none

CHIP-SPV · Dec 5, 2024 · 96835b0 · 96835b0
1 parent ffc06b1
commit 96835b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/CHIPBackend.hh b/src/CHIPBackend.hh
@@ -2150,6 +2150,7 @@ public:
    */
   template <class GraphNodeType, class... ArgTypes>
   bool captureIntoGraph(ArgTypes... ArgsPack) {
+    assert(getCaptureStatus() == hipStreamCaptureStatusNone);
     if (getCaptureStatus() == hipStreamCaptureStatusActive) {
       auto Graph = getCaptureGraph();
       auto Node = new GraphNodeType(ArgsPack...);
@@ -2165,7 +2166,8 @@ public:
 
   hipStreamCaptureStatus getCaptureStatus() const { return CaptureStatus_; }
   void setCaptureStatus(hipStreamCaptureStatus CaptureMode) {
-    CaptureStatus_ = CaptureMode;
+    assert(CaptureStatus_ == hipStreamCaptureStatusNone);
+    // CaptureStatus_ = CaptureMode;
   }
   hipStreamCaptureMode getCaptureMode() const { return CaptureMode_; }
   void setCaptureMode(hipStreamCaptureMode CaptureMode) {

diff --git a/src/CHIPBindings.cc b/src/CHIPBindings.cc
@@ -2532,6 +2532,7 @@ static inline hipError_t hipMemcpyAsyncInternal(void *Dst, const void *Src,
   NULLCHECK(Dst, Src);
 
   auto ChipQueue = Backend->findQueue(static_cast<chipstar::Queue *>(Stream));
+  assert(ChipQueue->getCaptureStatus() == hipStreamCaptureStatusNone);
   checkMemcpyKind(*ChipQueue->getDevice(), Kind);
   LOCK(ChipQueue->QueueMtx);
 
@@ -3322,6 +3323,7 @@ hipStreamCreateWithPriorityInternal(hipStream_t *Stream, unsigned int Flags,
   auto ClampedPriority = std::min(MinPriority, std::max(MaxPriority, Priority));
   chipstar::Queue *ChipQueue =
       Dev->createQueueAndRegister(FlagsParsed, ClampedPriority);
+  assert(ChipQueue->getCaptureStatus() == hipStreamCaptureStatusNone);
   *Stream = ChipQueue;
   return hipSuccess;
 }