From 96835b02576d9ce214922dd49df2cbe873b83a1c Mon Sep 17 00:00:00 2001
From: Paulius Velesko <pvelesko@pglc.io>
Date: Thu, 5 Dec 2024 18:55:51 +0200
Subject: [PATCH] TODO: capture Status none

---
 src/CHIPBackend.hh  | 4 +++-
 src/CHIPBindings.cc | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/CHIPBackend.hh b/src/CHIPBackend.hh
index efcb0b989..03ac6d95f 100644
--- 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
index 1aeaca886..cec8dff2b 100755
--- 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;
 }