Method EnqueueSegment

hipamd/src/hip_graph_internal.cpp:1580–1684

================================================================================================ Graph segment to queue dispatch matching

// ================================================================================================
// Graph segment to queue dispatch matching
hipError_t GraphExec::EnqueueSegment(const Segment& segment, hip::Stream* stream,
                                     amd::AccumulateCommand* accumulate) {
  hipError_t status = hipSuccess;

  // Find the SegmentBatch for this segment using O(1) map lookup
  SegmentBatch* segBatch = nullptr;
  auto segBatchIt = segmentBatches_.find(segment.id);
  if (segBatchIt != segmentBatches_.end()) {
    segBatch = &segBatchIt->second;
  }

  size_t batchIndex = 0;

  // Handle child graph segments - recursively enqueue the entire child graph
  if (segment.child_graph_ptr != nullptr) {
    auto childGraphExec = dynamic_cast<GraphExec*>(segment.child_graph_ptr);
    if (childGraphExec != nullptr) {
      // Child graphs share the same kernel arg manager as parent (for packet capture)
      if (childGraphExec->GetKernelArgManager() == nullptr) {
        auto kernArgMgr = GetKernelArgManager();
        if (kernArgMgr != nullptr) {
          kernArgMgr->retain();  // Increment ref count for child's reference
          childGraphExec->SetKernelArgManager(kernArgMgr);
        }
      }

      // Recursively enqueue the child graph with its own dependency tracking
      // Child graphs use their own parallel_streams_, so pass empty vector
      hipError_t child_status = hipSuccess;
      amd::Command* child_last_cmd = childGraphExec->EnqueueSegmentedGraph(
          stream, {}, &child_status);

      if (child_status != hipSuccess) {
        ClPrint(amd::LOG_ERROR, amd::LOG_CODE,
                "[hipGraph] EnqueueSegment: Failed to enqueue child graph, status=%d",
                child_status);
        return child_status;
      }

      // Child graph's work is already enqueued to the stream
      // The returned last command tracks completion - release our reference
      if (child_last_cmd != nullptr) {
        child_last_cmd->release();
      }
    }

    // Child graph segment has no regular nodes to process
    return hipSuccess;
  }

  // Process all nodes in this segment
  for (size_t i = 0; i < segment.nodes.size(); ++i) {
    auto& node = segment.nodes[i];
    if (DEBUG_HIP_GRAPH_DOT_PRINT) {
      node->stream_id_ = stream->GetStreamId();
    }
    if (!node->GraphCaptureEnabled()) {
      // Node doesn't support capture - execute individually